Skip to content

Commit 128c3a3

Browse files
committed
require tf=2.0.1 due to vulnerability in 2.0
1 parent a6dfe4f commit 128c3a3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
# Needed to actually package something
1717
packages= ['ct_segnet', 'ct_segnet.data_utils', 'ct_segnet.model_utils'],
1818
# Needed for dependencies
19-
install_requires=['tensorflow-gpu==2.0',
19+
install_requires=['tensorflow-gpu==2.0.1',
2020
'numpy', 'scikit-image', 'matplotlib', 'opencv-python', 'pandas', 'tifffile', 'scipy','scikit-learn','ConfigArgParse', 'tqdm'],
2121
version='1.2',
2222
license='BSD',

0 commit comments

Comments
 (0)