correct type hint

davidwilby · Jan 27, 2025 · 9765787 · 9765787
1 parent b0459e8
commit 9765787
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepsensor/model/model.py b/deepsensor/model/model.py
@@ -904,7 +904,7 @@ def stitch_clipped_predictions(
             patches_per_row: int,
             x1_ascend: bool = True,
             x2_ascend: bool = True,
-        ) -> dict:
+        ) -> Prediction:
             """Stitch patchwise predictions to form prediction at original extent.
 
             Parameters