Skip to content

Commit e5c0f07

Browse files
committed
Move over dependencies
1 parent eff2702 commit e5c0f07

File tree

2 files changed

+69
-68
lines changed

2 files changed

+69
-68
lines changed

pyproject.toml

+68
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,20 @@ classifiers = [
1717
]
1818
requires-python = "'>=3.8,<3.12"
1919
keywords = ['sdmetrics', 'sdmetrics', 'SDMetrics']
20+
dependencies = [
21+
"numpy>=1.20.0,<2;python_version<'3.10'",
22+
"numpy>=1.23.3,<2;python_version>='3.10'",
23+
"pandas>=1.1.3;python_version<'3.10'",
24+
"pandas>=1.3.4;python_version>='3.10' and python_version<'3.11'",
25+
"pandas>=1.5.0;python_version>='3.11'",
26+
"scikit-learn>=0.24,<2;python_version<='3.10'",
27+
"scikit-learn>=1.1.3,<2;python_version>='3.11'",
28+
"scipy>=1.5.4,<2;python_version<'3.10'",
29+
"scipy>=1.9.2,<2;python_version>='3.10'",
30+
'copulas>=0.9.0,<0.10',
31+
'tqdm>=4.15,<5',
32+
'plotly>=5.10.0,<6',
33+
]
2034

2135
[project.urls]
2236
"Source Code" = "https://github.com/sdv-dev/SDMetrics"
@@ -31,11 +45,65 @@ build-backend = "setuptools.build_meta"
3145
# Development dependencies
3246
[project.optional-dependencies]
3347

48+
torch = [
49+
"torch>=1.8.0;python_version<'3.10'",
50+
"torch>=1.11.0;python_version>='3.10' and python_version<'3.11'",
51+
"torch>=2.0.0;python_version>='3.11'",
52+
]
53+
54+
pomegranate = ['pomegranate>=0.14.1,<0.14.7']
55+
3456
test = [
57+
'sdmetrics[torch]',
3558
'pytest>=6.2.5,<7',
3659
'pytest-cov>=2.6.0,<3',
3760
'pytest-rerunfailures>=10',
3861
'jupyter>=1.0.0,<2',
3962
'rundoc>=0.4.3,<0.5',
4063
'invoke',
4164
]
65+
66+
dev = [
67+
'sdmetrics[test, torch]',
68+
# general
69+
'bumpversion>=0.5.3,<0.6',
70+
'pip>=9.0.1',
71+
'watchdog>=0.8.3,<0.11',
72+
73+
# style check
74+
'flake8>=3.7.7,<4',
75+
'flake8-absolute-import>=1.0,<2',
76+
'isort>=4.3.4,<5',
77+
'flake8-variables-names>=0.0.4,<0.1',
78+
'pydocstyle>=6.1.1,<6.2',
79+
'flake8-sfs>=0.0.3,<0.1',
80+
'flake8-docstrings>=1.5.0,<2',
81+
'dlint>=0.11.0,<0.12', # code security addon for flake8
82+
'pandas-vet>=0.2.2,<0.3',
83+
'pep8-naming>=0.12.1,<0.13',
84+
'flake8-pytest-style>=1.5.0,<2',
85+
'flake8-builtins>=1.5.3,<1.6',
86+
'flake8-comprehensions>=3.6.1,<3.7',
87+
'flake8-debugger>=4.0.0,<4.1',
88+
'flake8-mock>=0.3,<0.4',
89+
'flake8-fixme>=1.1.1,<1.2',
90+
'flake8-eradicate>=1.1.0,<1.2',
91+
'flake8-mutable>=1.2.0,<1.3',
92+
'flake8-print>=4.0.0,<4.1',
93+
'flake8-expression-complexity>=0.0.9,<0.1',
94+
'flake8-multiline-containers>=0.0.18,<0.1',
95+
'flake8-quotes>=3.3.0,<4',
96+
97+
# fix style issues
98+
'autoflake>=1.1,<2',
99+
'autopep8>=1.4.3,<1.6',
100+
101+
# distribute on PyPI
102+
'packaging>=20,<21',
103+
'twine>=1.10.0,<4',
104+
'wheel>=0.30.0',
105+
106+
# Advanced testing
107+
'coverage>=4.5.1,<6',
108+
'tox>=2.9.1,<4',
109+
]

setup.py

+1-68
Original file line numberDiff line numberDiff line change
@@ -11,79 +11,12 @@
1111
with open('HISTORY.md', encoding='utf-8') as history_file:
1212
history = history_file.read()
1313

14-
install_requires = [
15-
"numpy>=1.20.0,<2;python_version<'3.10'",
16-
"numpy>=1.23.3,<2;python_version>='3.10'",
17-
"pandas>=1.1.3;python_version<'3.10'",
18-
"pandas>=1.3.4;python_version>='3.10' and python_version<'3.11'",
19-
"pandas>=1.5.0;python_version>='3.11'",
20-
"scikit-learn>=0.24,<2;python_version<='3.10'",
21-
"scikit-learn>=1.1.3,<2;python_version>='3.11'",
22-
"scipy>=1.5.4,<2;python_version<'3.10'",
23-
"scipy>=1.9.2,<2;python_version>='3.10'",
24-
'copulas>=0.9.0,<0.10',
25-
'tqdm>=4.15,<5',
26-
'plotly>=5.10.0,<6',
27-
]
28-
29-
pomegranate_requires = [
30-
'pomegranate>=0.14.1,<0.14.7',
31-
]
32-
33-
torch_requires = [
34-
"torch>=1.8.0;python_version<'3.10'",
35-
"torch>=1.11.0;python_version>='3.10' and python_version<'3.11'",
36-
"torch>=2.0.0;python_version>='3.11'",
37-
]
14+
pomegranate_requires =
3815

3916
setup_requires = [
4017
'pytest-runner>=2.11.1',
4118
]
4219

43-
development_requires = [
44-
# general
45-
'bumpversion>=0.5.3,<0.6',
46-
'pip>=9.0.1',
47-
'watchdog>=0.8.3,<0.11',
48-
49-
# style check
50-
'flake8>=3.7.7,<4',
51-
'flake8-absolute-import>=1.0,<2',
52-
'isort>=4.3.4,<5',
53-
'flake8-variables-names>=0.0.4,<0.1',
54-
'pydocstyle>=6.1.1,<6.2',
55-
'flake8-sfs>=0.0.3,<0.1',
56-
'flake8-docstrings>=1.5.0,<2',
57-
'dlint>=0.11.0,<0.12', # code security addon for flake8
58-
'pandas-vet>=0.2.2,<0.3',
59-
'pep8-naming>=0.12.1,<0.13',
60-
'flake8-pytest-style>=1.5.0,<2',
61-
'flake8-builtins>=1.5.3,<1.6',
62-
'flake8-comprehensions>=3.6.1,<3.7',
63-
'flake8-debugger>=4.0.0,<4.1',
64-
'flake8-mock>=0.3,<0.4',
65-
'flake8-fixme>=1.1.1,<1.2',
66-
'flake8-eradicate>=1.1.0,<1.2',
67-
'flake8-mutable>=1.2.0,<1.3',
68-
'flake8-print>=4.0.0,<4.1',
69-
'flake8-expression-complexity>=0.0.9,<0.1',
70-
'flake8-multiline-containers>=0.0.18,<0.1',
71-
'flake8-quotes>=3.3.0,<4',
72-
73-
# fix style issues
74-
'autoflake>=1.1,<2',
75-
'autopep8>=1.4.3,<1.6',
76-
77-
# distribute on PyPI
78-
'packaging>=20,<21',
79-
'twine>=1.10.0,<4',
80-
'wheel>=0.30.0',
81-
82-
# Advanced testing
83-
'coverage>=4.5.1,<6',
84-
'tox>=2.9.1,<4',
85-
]
86-
8720
setup(
8821
extras_require={
8922
'test': tests_require + torch_requires,

0 commit comments

Comments
 (0)