Update Frequency Transform

.github/workflows/pythonapp.yml

@@ -103,9 +103,8 @@ jobs:
     - name: Install the dependencies
       run: |
         python -m pip install --user --upgrade pip wheel
-        python -m pip install torch==2.5.1 torchvision==0.20.1
         cat "requirements-dev.txt"
-        python -m pip install -r requirements-dev.txt
+        python -m pip install torch==2.5.1 torchvision==0.20.1 -r requirements-dev.txt
         python -m pip list
         python setup.py develop  # test no compile installation
       shell: bash

monai/transforms/signal/array.py

@@ -411,10 +411,11 @@ def __call__(self, signal: np.ndarray) -> Any:
         Args:
             signal: signal to be frequency removed
         """
+        signal = convert_to_tensor(signal, dtype=torch.double)
         b_notch, a_notch = convert_to_tensor(
-            iirnotch(self.frequency, self.quality_factor, self.sampling_freq), dtype=torch.float
+            iirnotch(self.frequency, self.quality_factor, self.sampling_freq), dtype=torch.double
         )
-        y_notched = filtfilt(convert_to_tensor(signal), a_notch, b_notch)
+        y_notched = filtfilt(signal, a_notch, b_notch)
 
         return y_notched
 

pyproject.toml

@@ -2,7 +2,7 @@
 requires = [
   "wheel",
   "setuptools",
-  "torch>=2.4.1",
+  "torch>=2.5.1",
   "ninja",
   "packaging"
 ]

requirements-dev.txt

@@ -1,6 +1,7 @@
 # Full requirements for developments
 -r requirements-min.txt
 pytorch-ignite==0.4.11
+torchaudio
 gdown>=4.7.3
 scipy>=1.12.0; python_version >= '3.9'
 itk>=5.2

requirements.txt

@@ -1,3 +1,3 @@
-torch>=2.4.1; platform_system != "Windows"
-torch>=2.4.1, !=2.7.0; platform_system == "Windows"
+torch>=2.5.1; platform_system != "Windows"
+torch>=2.5.1, !=2.7.0; platform_system == "Windows"
 numpy>=1.24,<3.0

setup.cfg

@@ -42,7 +42,7 @@ setup_requires =
     ninja
     packaging
 install_requires =
-    torch>=2.4.1
+    torch>=2.5.1
     numpy>=1.24,<3.0
 
 [options.extras_require]
@@ -108,6 +108,8 @@ torchio =
     torchio
 torchvision =
     torchvision
+torchaudio =
+    torchaudio
 itk =
     itk>=5.2
 tqdm =