Reverse add to_xarray since it can't hold spectrograms with time dependent

energy table

Signed-off-by: Alexis Jeandet <[email protected]>

Author: jeandet

pyistp/_impl.py

@@ -103,5 +103,5 @@ def _update_data_vars_lis(self):
                 if (var_type == 'data' or param_type == 'data') and not self.cdf.is_char(var):
                     self.data_variables.append(var)
 
-    def data_variable(self, var_name):
+    def data_variable(self, var_name) -> DataVariable:
         return _load_data_var(self.cdf, var_name)

pyistp/data_variable.py

@@ -1,5 +1,3 @@
-from xarray import DataArray
-
 
 class DataVariable:
     __slots__ = ("name", "values", 'attributes', 'axes', 'labels')
@@ -13,21 +11,6 @@ def __init__(self, name, values, attributes, axes, labels=None):
 
     def __len__(self):
         return len(self.values)
-
-    def to_xarray(self) -> DataArray:
-        axes = {ax.name: ax.values for ax in self.axes}
-        dims = [ax.name for ax in self.axes]
-        if len(self.values.shape) == 2 and len(axes) == 1:
-            axes['components'] = self.labels
-            dims.append('components')
-        return DataArray(
-            data=self.values,
-            dims=dims,
-            coords=axes,
-            name=self.name,
-            attrs=self.attributes
-        )
-
     def __repr__(self):
         return f"""DataVariable: {self.name}
 Attributes:

pyistp/loader.py

@@ -1,5 +1,5 @@
 from typing import List
-
+from .data_variable import DataVariable
 
 class ISTPLoader:
 
@@ -16,5 +16,5 @@ def attribute(self, key):
     def data_variables(self) -> List[str]:
         return self._impl.data_variables
 
-    def data_variable(self, var_name) -> List[str]:
+    def data_variable(self, var_name) -> DataVariable:
         return self._impl.data_variable(var_name)