Merge pull request #59 from JesseLivezey/requirements

JesseLivezey · web-flow · commit 3629e9aa8b17 · 2021-05-18T15:59:23.000-07:00
Requirements
diff --git a/dca/plotting/fig1.py b/dca/plotting/fig1.py
@@ -5,7 +5,7 @@
 from matplotlib.patches import Ellipse
 from mpl_toolkits.mplot3d import Axes3D  # noqa: F401
 
-from . import style
+from .. import style
 
 
 dim_colors = ["red", "coral", "gray", "black"]
diff --git a/environment.yml b/environment.yml
@@ -7,5 +7,5 @@ dependencies:
   - pandas
   - scikit-learn
   - matplotlib
-  - pytorch
+  - pytorch>=1.7.0
   - cpuonly
diff --git a/requirements.txt b/requirements.txt
@@ -5,4 +5,4 @@ pandas
 scikit-learn
 matplotlib
 -f https://download.pytorch.org/whl/torch_stable.html
-torch==1.3.1+cpu
+torch>=1.7.0+cpu
