Adding Backend to fieldset repr

To show which fields are WindowedArrays, which are Dask and which are NumPy

Author: erikvansebille

src/parcels/_reprs.py

@@ -8,7 +8,9 @@
 
 import numpy as np
 import xarray as xr
+from dask.base import is_dask_collection
 
+from parcels._core._windowed_array import WindowedArray
 from parcels._python import isinstance_noimport
 
 if TYPE_CHECKING:
@@ -190,13 +192,15 @@ class _FieldSetDescriptionRow:
     model_id: int | None
     name: str
     interp_method_or_value: str
+    backend: str | None = None
 
     def to_dict(self) -> dict[str, str]:
         return {
             "Name": self.name,
             "Type": self.type_,
             "Grid number": str(self.model_id) if self.model_id is not None else "-",
             "Interp method / value": self.interp_method_or_value,
+            "Backend": self.backend if self.backend is not None else "-",
         }
 
 
@@ -213,6 +217,20 @@ def _print_time_interval(time_interval: TimeInterval | None) -> str:
     return repr((time_interval.left, time_interval.right))
 
 
+def _field_backend(field: Field | VectorField) -> str | None:
+    if hasattr(field, "data"):
+        if isinstance(field.data, WindowedArray):
+            return "WindowedArray"
+        elif is_dask_collection(field.data.data):
+            return "Dask"
+        elif isinstance(field.data.data, np.ndarray):
+            return "NumPy"
+        else:
+            return type(field.data).__name__
+    else:
+        return None
+
+
 def fieldset_describe(fieldset: FieldSet) -> str:
     rows: list[_FieldSetDescriptionRow] = []
     models: dict[int, int] = {}  # mapping of memory ID to a human readable ID
@@ -235,6 +253,7 @@ def fieldset_describe(fieldset: FieldSet) -> str:
                 model_id=model_id,
                 name=field.name,
                 interp_method_or_value=repr(field.interp_method),
+                backend=_field_backend(field),
             )
         )
     for k, v in fieldset.context.items():
@@ -244,6 +263,7 @@ def fieldset_describe(fieldset: FieldSet) -> str:
                 model_id=None,
                 name=k,
                 interp_method_or_value=repr(v),
+                backend=None,
             )
         )
     return (