setup.py

# Copyright 2019-2023 The AmpliGraph Authors. All Rights Reserved.
#
# This file is Licensed under the Apache License, Version 2.0.
# A copy of the Licence is available in LICENCE, or at:
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
from setuptools import setup, find_packages
from ampligraph import __version__ as version

setup_params = dict(name='ampligraph',
                    version=version,
                    description='A Python library for relational learning on knowledge graphs.',
                    url='https://github.com/Accenture/AmpliGraph/',
                    author='Accenture Dublin Labs',
                    author_email='about@ampligraph.org',
                    license='Apache 2.0',
                    packages=find_packages(exclude=('tests', 'docs')),
                    include_package_data=True,
                    zip_safe=False,
                    install_requires=[
                        'numpy>=1.14.3',
                        'pytest>=3.5.1',
                        'scikit-learn>=0.19.1',
                        'tqdm>=4.23.4',
                        'pandas>=0.23.1',
                        'sphinx==5.0.2',
                        'myst-parser==0.18.0',
                        'docutils<0.18',
                        'sphinx_rtd_theme==1.0.0',
                        'sphinxcontrib-bibtex==2.4.2',
                        'beautifultable>=0.7.0',
                        'pyyaml>=3.13',
                        'rdflib>=4.2.2',
                        'scipy==1.10.0',
                        'networkx>=2.3',
                        'flake8>=3.7.7',
                        'setuptools>=36',
                        'matplotlib>=3.7',
                        'docopt==0.6.2',
                        'schema==0.7.5'
                    ],
                    setup_requires=['pytest-runner'],
                    tests_require=['pytest', 
                                   'pytest_mock>=3.10.0',
                                   'mock>=5.0.1'])

if __name__ == '__main__':
    setup(**setup_params)