small correction

mindee · Nov 22, 2024 · 51b058c · 51b058c
1 parent 6eb3455
commit 51b058c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/doctr/models/detection/predictor/pytorch.py b/doctr/models/detection/predictor/pytorch.py
@@ -39,7 +39,7 @@ def forward(
         pages: list[np.ndarray | torch.Tensor],
         return_maps: bool = False,
         **kwargs: Any,
-    ) -> list[dict[str, np.ndarray]] | tuple[list[dict[str, np.ndarray]] | list[np.ndarray]]:
+    ) -> list[dict[str, np.ndarray]] | tuple[list[dict[str, np.ndarray]], list[np.ndarray]]:
         # Extract parameters from the preprocessor
         preserve_aspect_ratio = self.pre_processor.resize.preserve_aspect_ratio
         symmetric_pad = self.pre_processor.resize.symmetric_pad

diff --git a/doctr/models/detection/predictor/tensorflow.py b/doctr/models/detection/predictor/tensorflow.py
@@ -39,7 +39,7 @@ def __call__(
         pages: list[np.ndarray | tf.Tensor],
         return_maps: bool = False,
         **kwargs: Any,
-    ) -> list[dict[str, np.ndarray]] | tuple[list[dict[str, np.ndarray]] | list[np.ndarray]]:
+    ) -> list[dict[str, np.ndarray]] | tuple[list[dict[str, np.ndarray]], list[np.ndarray]]:
         # Extract parameters from the preprocessor
         preserve_aspect_ratio = self.pre_processor.resize.preserve_aspect_ratio
         symmetric_pad = self.pre_processor.resize.symmetric_pad