p-kit/setup.py at main · IBM/p-kit · 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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
import os.path as op

from setuptools import setup, find_packages


# get the version (don't import mne here, so dependencies are not needed)
version = None
with open(op.join('p_kit', '_version.py'), 'r') as fid:
    for line in (line.strip() for line in fid):
        if line.startswith('__version__'):
            version = line.split('=')[1].strip().strip('"')
            break
if version is None:
    raise RuntimeError('Could not determine version')

with open('README.md', 'r', encoding="utf8") as fid:
    long_description = fid.read()

setup(name='p-kit',
      version=version,
      description='Probabilistic circuit simulator',
      url='https://github.com/IBM/p-kit',
      author='Gregoire Cattan, Anton Andreev',
      author_email='gregoire.cattan@ibm.com',
      license='BSD (3-clause)',
      packages=find_packages(),
      long_description=long_description,
      long_description_content_type='text/markdown',
      project_urls={
          'Documentation': 'https://github.com/IBM/p-kit/wiki',
          'Source': 'https://github.com/IBM/p-kit',
          'Tracker': 'https://github.com/IBM/p-kit/issues/',
      },
      platforms='any',
      python_requires=">=3.10",
      install_requires=[
                        'numpy<2.3',
                        'cython==3.2.5',
                        'cvxpy==1.7.5',
                        'scipy==1.15.3',
                        'matplotlib==3.10.9',
                        'networkx',
                        'joblib'
                        ],
      extras_require={
                      'tests': ['pytest', 'seaborn', 'flake8'],
                      'gpu': ['cupy-cuda13x'],
                      'docplex': ['docplex'],
                      },
      zip_safe=False,
)