fix: fix mypy check

Signed-off-by: anna-charlotte <[email protected]>
docarray · JoanFM · Jan 3, 2023 · Dec 14, 2022 · Dec 14, 2022 · Dec 15, 2022
commit 71af63011c87825323e63c08f3da26f50a85f55c
diff --git a/docarray/computation/numpy_backend.py b/docarray/computation/numpy_backend.py
@@ -41,7 +41,7 @@ def stack(
         return np.stack(tensors, axis=dim)
 
     @staticmethod
-    def n_dim(array: np.ndarray) -> int:
+    def n_dim(array: 'np.ndarray') -> int:
         return array.ndim
 
     class Retrieval(AbstractComputationalBackend.Retrieval[np.ndarray]):

diff --git a/docarray/computation/torch_backend.py b/docarray/computation/torch_backend.py
@@ -40,7 +40,7 @@ def stack(
         return torch.stack(tensors, dim=dim)
 
     @staticmethod
-    def n_dim(array: torch.Tensor) -> int:
+    def n_dim(array: 'torch.Tensor') -> int:
         return array.ndim
 
     class Retrieval(AbstractComputationalBackend.Retrieval[torch.Tensor]):

diff --git a/docarray/typing/tensor/audio/abstract_audio_tensor.py b/docarray/typing/tensor/audio/abstract_audio_tensor.py
@@ -29,7 +29,8 @@ def save_to_wav_file(
         :param sample_rate: sampling frequency
         :param sample_width: sample width in bytes
         """
-        n_channels = 2 if self.get_comp_backend().n_dim(array=self) > 1 else 1
+        comp_backend = self.get_comp_backend()
+        n_channels = 2 if comp_backend.n_dim(array=self) > 1 else 1  # type: ignore
 
         with wave.open(file_path, 'w') as f:
             f.setnchannels(n_channels)