Better solution for installing patch versions; add update to utility …

…functions in 'test_module.py'
fkodom · Jan 28, 2022 · c3a5260 · c3a5260
1 parent 42dcb4e
commit c3a5260
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml
@@ -13,7 +13,7 @@ jobs:
     strategy:
       matrix:
         python: ["3.7", "3.8", "3.9"]
-        torch: ["1.7.*", "1.8.*", "1.9.*", "1.10.*"]
+        torch: ["1.7", "1.8", "1.9", "1.10"]
 
     steps:
       - name: Checkout
@@ -26,7 +26,7 @@ jobs:
 
       - name: Install Package
         run: |
-          pip install torch==${{ matrix.torch }}
+          pip install torch~=${{ matrix.torch }}.0
           pip install .[test]
 
       - name: Test

diff --git a/tests/test_module.py b/tests/test_module.py
@@ -5,7 +5,7 @@
 import torch.nn.functional as f
 
 from fft_conv_pytorch.fft_conv import _FFTConv
-from tests.utils import _assert_almost_equal, _gcd
+from fft_conv_pytorch.utils import _assert_almost_equal, _gcd
 
 
 @pytest.mark.parametrize("in_channels", [2, 3])