Test with "setup.py test"

doodleima · Oct 7, 2019 · 7b0c430 · 7b0c430
1 parent ed63cc4
commit 7b0c430
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -30,6 +30,8 @@
     zip_safe=False,
     packages=['torchlars'],
     install_requires=['torch'],
+    setup_requires=['pytest-runner'],
+    tests_require=['pytest>=4'],
     classifiers=[
         'Development Status :: 3 - Alpha',
         'Intended Audience :: Science/Research',