Update _extract_var_start_and_end_years() to return ints

E3SM-Project · Oct 28, 2024 · 40de792 · 40de792
1 parent 648e1e1
commit 40de792
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/e3sm_diags/driver/utils/dataset_xr.py b/e3sm_diags/driver/utils/dataset_xr.py
@@ -1245,7 +1245,7 @@ def _extract_var_start_and_end_years(self, ds: xr.Dataset) -> Tuple[int, int]:
         if time_coords[-1].dt.month == 1:
             end_year -= 1
 
-        return start_year, end_year
+        return start_year.item(), end_year.item()
 
     def _get_slice_with_bounds(
         self, ds: xr.Dataset, year_str: str, slice_type: Literal["start", "end"]