Fix setup.py

sovrasov · Dec 26, 2023 · 091ad0f · 091ad0f
1 parent d7e1ad9
commit 091ad0f
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/setup.py b/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',