Merge pull request #130 from sovrasov/fix_setup

Fix setup

Author: sovrasov

setup.py

@@ -10,18 +10,22 @@
 
 from setuptools import find_packages, setup
 
-VERSION = '0.7.2'
+VERSION = '0.7.2.1'
 
 requirements = [
     'torch',
 ]
 
 SETUP_DIR = Path(__file__).resolve().parent
-
-TEST_BASE_EXTRAS = (SETUP_DIR / 'test_requirements.txt').read_text()
-EXTRAS_REQUIRE = {
-    'dev': TEST_BASE_EXTRAS,
-}
+TEST_REQ = SETUP_DIR / 'test_requirements.txt'
+
+if TEST_REQ.is_file():
+    TEST_BASE_EXTRAS = TEST_REQ.read_text()
+    EXTRAS_REQUIRE = {
+        'dev': TEST_BASE_EXTRAS,
+    }
+else:
+    EXTRAS_REQUIRE = {}
 
 setup(
     name='ptflops',