fix numba typing issue

Deltares · Jun 29, 2023 · 9e5db8b · 9e5db8b
1 parent 46fab5e
commit 9e5db8b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/xugrid/regrid/regridder.py b/xugrid/regrid/regridder.py
@@ -235,8 +235,8 @@ def _csr_from_dataset(dataset: xr.Dataset) -> WeightMatrixCSR:
             dataset["__regrid_data"].values,
             dataset["__regrid_indices"].values,
             dataset["__regrid_indptr"].values,
-            dataset["__regrid_n"].values,
-            dataset["__regrid_nnz"].values,
+            dataset["__regrid_n"].values[()],
+            dataset["__regrid_nnz"].values[()],
         )
 
     @staticmethod
@@ -245,7 +245,7 @@ def _coo_from_dataset(dataset: xr.Dataset) -> WeightMatrixCOO:
             dataset["__regrid_data"].values,
             dataset["__regrid_row"].values,
             dataset["__regrid_col"].values,
-            dataset["__regrid_nnz"].values,
+            dataset["__regrid_nnz"].values[()],
         )
 
     @abc.abstractclassmethod