Added torch to the main env

ai2es · Aug 15, 2024 · 6d4d338 · 6d4d338
1 parent e8139ec
commit 6d4d338
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -53,7 +53,7 @@ The package contains three scripts for training three regression models, and one
 The regression examples are trained on our surface layer ("SL") dataset for predicting latent heat and other quantities, 
 and the categorical example is trained on a precipitation dataset ("p-type").
 
-The law of total variance for each model prediction target may be computed as 
+The law of total variance for each model prediction target may be computed as
 
 $$LoTV = E[\sigma^2] + Var[\mu]$$ 
 

diff --git a/environment.yml b/environment.yml
@@ -27,6 +27,7 @@ dependencies:
   - pip:
     - tensorflow
     - keras
+    - torch
     - echo-opt
     - hagelslag
     - bridgescaler