tighten return type

v0lta · Jun 3, 2024 · c2dccd0 · c2dccd0
1 parent 310b1af
commit c2dccd0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ptwt/packets.py b/src/ptwt/packets.py
@@ -183,7 +183,7 @@ def _get_wavedec(
     def _get_waverec(
         self,
         length: int,
-    ) -> Callable[[list[torch.Tensor]], torch.Tensor]:
+    ) -> Callable[[Sequence[torch.Tensor]], torch.Tensor]:
         if self.mode == "boundary":
             if length not in self._matrix_waverec_dict.keys():
                 self._matrix_waverec_dict[length] = MatrixWaverec(