update docs conda env

RMeli · Aug 1, 2022 · 9f9385f · 9f9385f
1 parent bb91a9c
commit 9f9385f
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/docs/requirements.yaml b/docs/requirements.yaml
@@ -1,12 +1,20 @@
 name: docs
 channels:
+ - conda-forge
+ - pytorch
 dependencies:
- # Base depends
  - python
  - pip
 
+ - numpy
 
+ - pytorch
+ - ignite
 
-  # Pip-only installs
- #- pip:
+ - mlflow
+ - scikit-learn>=1.0
 
+ - tqdm
+
+ - pip:
+ - molgrid