be able to load Metadata with additional columns

attila-balint-kul · Feb 20, 2024 · e6a98bd · e6a98bd
1 parent f21ac63
commit e6a98bd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/enfobench/__version__.py b/src/enfobench/__version__.py
@@ -1 +1 @@
-__version__ = "0.4.0"
+__version__ = "0.4.1"
diff --git a/src/enfobench/dataset/_dataset.py b/src/enfobench/dataset/_dataset.py
@@ -72,7 +72,8 @@ def get_by_unique_id(self, unique_id: str) -> Metadata:
         if df.empty:
             msg = f"Unique id '{unique_id}' was not found."
             raise KeyError(msg)
-        return Metadata(**df.to_dict(orient="records")[0])
+        metadata = df.iloc[0].to_dict()
+        return Metadata(**{k: v for k, v in metadata.items() if k in Metadata.__annotations__.keys()})
 
 
 class WeatherSubset(Subset):