update contrastive deps to fix units

Signed-off-by: Giridhar Ganapavarapu <[email protected]>
Trusted-AI · May 1, 2023 · cdd1e62 · cdd1e62
1 parent b75c02c
commit cdd1e62
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -64,6 +64,9 @@
         "tensorflow==1.14",
         "requests",
         "scipy>=0.17",
+        "scikit-image",
+        "torch",
+        "h5py<3.0.0",  # to resolve keras error: 'str' object has no attribute 'decode'
     ],
     "shap": [
         "keras==2.3.1",