ad PyTorch to setup.py

maxkeller321 · maxkeller321 · commit 1d77cdef0f30 · 2021-06-08T09:13:41.000+02:00
diff --git a/setup.py b/setup.py
@@ -10,5 +10,5 @@
     url='https://modAL-python.github.io/',
     packages=['modAL', 'modAL.models', 'modAL.utils'],
     classifiers=['Development Status :: 4 - Beta'],
-    install_requires=['numpy==1.20.0', 'scikit-learn>=0.18', 'scipy>=0.18', 'pandas>=1.1.0', 'skorch==0.9.0'],
+    install_requires=['numpy==1.20.0', 'scikit-learn>=0.18', 'scipy>=0.18', 'pandas>=1.1.0', 'skorch==0.9.0', 'torch>=1.8.1'],
 )

Original file line number	Diff line number	Diff line change
`@@ -10,5 +10,5 @@`
`10`	`10`	`url='https://modAL-python.github.io/',`
`11`	`11`	`packages=['modAL', 'modAL.models', 'modAL.utils'],`
`12`	`12`	`classifiers=['Development Status :: 4 - Beta'],`
`13`		`- install_requires=['numpy==1.20.0', 'scikit-learn>=0.18', 'scipy>=0.18', 'pandas>=1.1.0', 'skorch==0.9.0'],`
	`13`	`+ install_requires=['numpy==1.20.0', 'scikit-learn>=0.18', 'scipy>=0.18', 'pandas>=1.1.0', 'skorch==0.9.0', 'torch>=1.8.1'],`
`14`	`14`	`)`