Update deepsensor/model/model.py

Co-authored-by: David Wilby <24752124+davidwilby@users.noreply.github.com>
davidwilby · Jan 21, 2025 · 9943e99 · 9943e99
1 parent 358b884
commit 9943e99
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepsensor/model/model.py b/deepsensor/model/model.py
@@ -810,7 +810,7 @@ def overlap_index(
                 ),
             )
 
-        def get_coordinate_extent(ds, x1_ascend, x2_ascend) -> tuple:
+        def get_coordinate_extent(ds: Union[xr.DataArray, xr.Dataset], x1_ascend: bool, x2_ascend: bool) -> tuple:
             """
             Get coordinate extent of dataset. This method is applied to either X_t or patchwise predictions.