From e6a98bd84f03ffbd37c98ec795e0fa355fe8013c Mon Sep 17 00:00:00 2001
From: Attila Balint <attila.balint@kuleuven.be>
Date: Tue, 20 Feb 2024 16:20:23 +0100
Subject: [PATCH] be able to load Metadata with additional columns

---
 src/enfobench/__version__.py      | 2 +-
 src/enfobench/dataset/_dataset.py | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

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):