Better solution for installing patch versions; add update to utility functions in 'test_module.py'

Author: fkodom

.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

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])