diff --git a/src/enfobench/__version__.py b/src/enfobench/__version__.py index 6a9beea..3d26edf 100644 --- 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 index 2ae40bf..6f18c48 100644 --- 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):