pytorch_geometric_edge/setup.py at master · pbielak/pytorch_geometric_edge · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
from setuptools import find_packages, setup

with open("README.md", "r") as fin:
    long_description = fin.read()


setup(
    name='torch_geometric_edge',
    version='0.0.1',
    url='',
    license='MIT',
    author='Piotr Bielak',
    author_email='piotr.bielak@pwr.edu.pl',
    description='Edge representation learning library',
    long_description=long_description,
    long_description_content_type='text/markdown',
    python_requires='>=3.8',
    install_requires=[
        'gensim>=4.2.0',
        'networkx>=2.8.5',
        'pandas>=1.4.3',
        'requests>=2.28.1',
        'scikit-learn>=1.1.1',
        'torch>=1.10.0',
        'torch-geometric==2.0.4',
    ],
    extras_require={
        'test': [
            'flake8',
            'isort',
            'pylint',
            'pytest',
        ],
    },
    packages=find_packages(),
)