KGDiff/setup.py at main · CMACH508/KGDiff · 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
'''
Author: QHGG
Date: 2023-10-13 13:58:29
LastEditTime: 2023-10-13 15:28:39
LastEditors: QHGG
Description:
FilePath: /KGDiff/setup.py
'''

__version__ = '0.1.2'
URL = 'https://github.com/CMACH508/KGDiff'

from setuptools import setup, find_packages

setup(name='KGDiff',
      version=__version__,
      description='KGDiff packages',
      url=URL,
      author='qhonearth',
      author_email='qhonearth@sjtu.edu.cn',
      license='MIT',
      download_url=f'{URL}/archive/{__version__}.tar.gz',
      keywords=['pytorch', 'KGDiff', 'drug design', 'diffusion model'],
    #   packages=['KGDiff'],
      packages=find_packages(),
      entry_points={
          'console_scripts': [
              'kg_gen = scripts.sample_diffusion:main',
              'kg_gen4poc = scripts.sample_for_pocket:main',
              'kg_train = scripts.train_diffusion:main',
              'kg_eval = scripts.evaluate_diffusion:main',
          ],
      },
      zip_safe=False)