Fixes typing issues that were resulting in mypy errors

Author: KevinShuman

clouddrift/ragged.py

@@ -787,11 +787,11 @@ def subset(
 
 
 def unpack(
-    ragged_array: np.ndarray,
-    rowsize: np.ndarray[int],
+    ragged_array: np.ndarray | xr.DataArray,
+    rowsize: np.ndarray | xr.DataArray,
     rows: int | Iterable[int] = None,
     axis: int = 0,
-) -> list[np.ndarray]:
+) -> list[np.ndarray | xr.DataArray]:
     """Unpack a ragged array into a list of regular arrays.
 
     Unpacking a ``np.ndarray`` ragged array is about 2 orders of magnitude

tests/kinematics_test.py

@@ -24,9 +24,9 @@ def sample_ragged_array() -> RaggedArray:
     latitude = [[-90, -45, 45, 90, 0], [10, 20], [10, 20, 30, 40]]
     t = [[1, 2, 3, 4, 5], [4, 5], [2, 3, 4, 5]]
     test = [
-        [True, True, True, False, False],
-        [False, False],
-        [True, False, False, False],
+        [1, 1, 1, 0, 0],
+        [0, 0],
+        [1, 0, 0, 0],
     ]
     rowsize = [len(x) for x in longitude]
     ids = [[d] * rowsize[i] for i, d in enumerate(drifter_id)]

tests/plotting_test.py

@@ -2,7 +2,7 @@
 import unittest
 from unittest.mock import patch
 
-import cartopy.crs as ccrs
+import cartopy.crs as ccrs  # type: ignore
 import matplotlib.pyplot as plt
 import numpy as np