remove type annotations.

CEA-COSMIC · Feb 19, 2024 · 3e138b8 · 3e138b8
1 parent c9f5ca1
commit 3e138b8
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/src/modopt/opt/linear/wavelet.py b/src/modopt/opt/linear/wavelet.py
@@ -285,10 +285,10 @@ class TorchWaveletTransform:
 
     def __init__(
         self,
-        shape: tuple[int, ...],
-        wavelet: str,
-        level: int,
-        mode: str,
+        shape,
+        wavelet,
+        level,
+        mode,
     ):
         self.wavelet = wavelet
         self.level = level
@@ -417,10 +417,10 @@ class CupyWaveletTransform(LinearParent):
 
     def __init__(
         self,
-        shape: tuple[int, ...],
-        wavelet: str,
-        level: int,
-        mode: str,
+        shape,
+        wavelet,
+        level,
+        mode,
     ):
         self.wavelet = wavelet
         self.level = level