Merge pull request #21 from cics-nd/issue-20

Issue 20

Author: sdatkinson

environment.yml

@@ -6,11 +6,11 @@ name: gptorch
 channels:
   - pytorch
 dependencies:
-  - pytorch
-  - numpy>=1.10
-  - scipy>=0.18
-  - matplotlib>=2.1.2
-  - scikit-learn>=0.19.1
-  - pytest>=3.5.0
   - graphviz>=0.9
   - jupyter
+  - matplotlib>=2.1.2
+  - numpy>=1.10
+  - pytest>=3.5.0
+  - pytorch
+  - scikit-learn>=0.19.1
+  - scipy>=0.18

gptorch/functions.py

@@ -10,8 +10,8 @@
 from torch.nn import functional as F
 import numpy as np
 
-torch_version = [int(s) for s in torch.__version__.split(".")]
-_potri = torch.cholesky_inverse if torch_version >= [1, 1, 0] else torch.potri
+_torch_majorminor = [int(s) for s in torch.__version__.split(".")[:2]]
+_potri = torch.cholesky_inverse if _torch_majorminor >= [1, 1] else torch.potri
 
 
 TRIANGULAR_SOLVE = "triangular_solve" in dir(torch)

requirements.txt

@@ -2,12 +2,12 @@
 # Created: Saturday, July 13th 2019, 6:32:27 pm
 # Author: Steven Atkinson (steven@atkinson.mn)
 
-torch>=1
-numpy>=1.10
-scipy>=0.18
+graphviz>=0.9
+jupyter
 matplotlib>=2.1.2
-scikit-learn>=0.19.1
+numpy>=1.10
 pytest>=3.5.0
 pytest-cov
-graphviz>=0.9
-jupyter
+scikit-learn>=0.19.1
+scipy>=0.18
+torch>=1

test/test_functions.py

@@ -0,0 +1,7 @@
+# File: test_functions.py
+# File Created: Saturday, 30th October 2021 11:06:35 am
+# Author: Steven Atkinson (steven@atkinson.mn)
+
+def test_importable():
+    from gptorch import functions
+