@@ -17,6 +17,20 @@ classifiers = [
17
17
]
18
18
requires-python = " '>=3.8,<3.12"
19
19
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
+ ]
20
34
21
35
[project .urls ]
22
36
"Source Code" = " https://github.com/sdv-dev/SDMetrics"
@@ -31,11 +45,65 @@ build-backend = "setuptools.build_meta"
31
45
# Development dependencies
32
46
[project .optional-dependencies ]
33
47
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
+
34
56
test = [
57
+ ' sdmetrics[torch]' ,
35
58
' pytest>=6.2.5,<7' ,
36
59
' pytest-cov>=2.6.0,<3' ,
37
60
' pytest-rerunfailures>=10' ,
38
61
' jupyter>=1.0.0,<2' ,
39
62
' rundoc>=0.4.3,<0.5' ,
40
63
' invoke' ,
41
64
]
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
+ ]
0 commit comments