Skip to content

Commit 4069e6f

Browse files
committed
Fix setup.py
1 parent a05c2a8 commit 4069e6f

File tree

1 file changed

+10
-6
lines changed

1 file changed

+10
-6
lines changed

setup.py

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -10,18 +10,22 @@
1010

1111
from setuptools import find_packages, setup
1212

13-
VERSION = '0.7.2'
13+
VERSION = '0.7.2.1'
1414

1515
requirements = [
1616
'torch',
1717
]
1818

1919
SETUP_DIR = Path(__file__).resolve().parent
20-
21-
TEST_BASE_EXTRAS = (SETUP_DIR / 'test_requirements.txt').read_text()
22-
EXTRAS_REQUIRE = {
23-
'dev': TEST_BASE_EXTRAS,
24-
}
20+
TEST_REQ = SETUP_DIR / 'test_requirements.txt'
21+
22+
if TEST_REQ.is_file():
23+
TEST_BASE_EXTRAS = TEST_REQ.read_text()
24+
EXTRAS_REQUIRE = {
25+
'dev': TEST_BASE_EXTRAS,
26+
}
27+
else:
28+
EXTRAS_REQUIRE = {}
2529

2630
setup(
2731
name='ptflops',

0 commit comments

Comments
 (0)