Merge pull request #385 from DHI/mikeio-dataset-no-longer-mapping

mikeio.Dataset is no longer a mapping...
DHI · Jan 8, 2024 · 1f1d02b · 1f1d02b
2 parents 64e17dd + 9fedfa0
commit 1f1d02b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/modelskill/model/dfsu.py b/modelskill/model/dfsu.py
@@ -189,8 +189,9 @@ def extract_track(self, observation: TrackObservation) -> TrackModelResult:
             ds_model.rename({self.sel_items.values: self.name}, inplace=True)
             aux_items = self.sel_items.aux
 
-        x_name = "Longitude" if "Longitude" in ds_model else "x"
-        y_name = "Latitude" if "Latitude" in ds_model else "y"
+        item_names = [i.name for i in ds_model.items]
+        x_name = "Longitude" if "Longitude" in item_names else "x"
+        y_name = "Latitude" if "Latitude" in item_names else "y"
 
         return TrackModelResult(
             data=ds_model.dropna(),  # TODO: not on aux cols