Maintenance

.github/workflows/ci.yml

@@ -16,11 +16,11 @@ jobs:
       max-parallel: 4
       matrix:
         python-version: [3.8]
-        pytorch-version: [1.11.0, 2.1.0]
+        pytorch-version: [1.11.0, 2.1.1]
 
     steps:
       - name: Clone
-        uses: actions/checkout@v3
+        uses: actions/checkout@v4
 
       - name: Python
         uses: actions/setup-python@v4

pyproject.toml

@@ -30,7 +30,7 @@
   dependencies = [
     "soundfile",
     "torch >= 1.11.0",
-    "torchcrepe >= 0.0.16, <= 0.0.18",
+    "torchcrepe >= 0.0.16",
     "torchlpc >= 0.2.0",
     "numpy",
     "vector-quantize-pytorch >= 0.8.0",
@@ -96,4 +96,5 @@
 [tool.pytest.ini_options]
   filterwarnings = [
     "ignore:Casting complex values to real discards the imaginary part",
+    "ignore:__floordiv__ is deprecated",
   ]

tests/conftest.py

@@ -21,3 +21,11 @@
 @pytest.fixture(scope="session", autouse=True)
 def scope_session():
     torch.set_default_dtype(torch.double)
+
+
+@pytest.fixture
+def float_test():
+    dtype = torch.get_default_dtype()
+    torch.set_default_dtype(torch.float)
+    yield
+    torch.set_default_dtype(dtype)

tests/test_pitch.py

@@ -23,7 +23,7 @@
 
 @pytest.mark.parametrize("device", ["cpu", "cuda"])
 @pytest.mark.parametrize("out_format", [0, 1, 2])
-def test_compatibility(device, out_format, P=80, sr=16000, L=80, H=180):
+def test_compatibility(float_test, device, out_format, P=80, sr=16000, L=80, H=180):
     pitch = diffsptk.Pitch(P, sr, f_min=L, f_max=H, out_format=out_format, model="tiny")
 
     def eq(o, sr):
@@ -73,6 +73,6 @@ def inner_eq(y_hat, y):
 
 @pytest.mark.parametrize("device", ["cpu", "cuda"])
 @pytest.mark.parametrize("out_format", ["prob", "embed"])
-def test_differentiable(device, out_format, P=80, sr=16000, B=2, T=1000):
+def test_differentiable(float_test, device, out_format, P=80, sr=16000, B=2, T=1000):
     pitch = diffsptk.Pitch(P, sr, out_format=out_format, model="tiny")
     U.check_differentiable(device, pitch, [B, T])