Skip to content

Commit 7417fc3

Browse files
authored
Update setup.py to use scikit-learn
sklearn is deprecated
1 parent a69cc22 commit 7417fc3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Diff for: setup.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
VERSION = '0.2'
1515

1616
# What packages are required for this module to be executed?
17-
REQUIRED = ['matplotlib', 'numpy', 'networkx', 'cvxpy', 'pandas', 'scipy', 'sklearn', 'colorlog']
17+
REQUIRED = ['matplotlib', 'numpy', 'networkx', 'cvxpy', 'pandas', 'scipy', 'scikit-learn', 'colorlog']
1818
EXTRAS = {'GPU': ['cupy']}
1919

2020
# The rest you shouldn't have to touch too much :)

0 commit comments

Comments
 (0)