Merge pull request #243 from kipoi/fix-ci

Pinned packages for AttentiveChrome
kipoi · Apr 1, 2021 · 55a03dc · 55a03dc
2 parents b7078c7 + 18e7b9a
commit 55a03dc
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/AttentiveChrome/model-template.yaml b/AttentiveChrome/model-template.yaml
@@ -31,10 +31,11 @@ info: # General information about the model
         - RNA expression
 dependencies:
     conda: # install via conda
-      - python==3.6.3
-      - numpy  
-      - pytorch::pytorch-cpu 
-      - pytorch::torchvision-cpu   
+      - python=3.6.3
+      - numpy=1.19.2
+      - pytorch::pytorch-cpu=1.3.1
+      - pytorch::torchvision-cpu=0.3.0
+      - pip=21.0.1  
 schema:  # Model schema
     inputs:
         shape: (100, 5)  # array shape of a single sample (omitting the batch dimension)