fix: add dataset name property to pseudo labels (#53)

tilman151 · Nov 17, 2023 · 2cf9fd0 · 2cf9fd0
1 parent c99d1e7
commit 2cf9fd0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/rul_adapt/approach/pseudo_labels.py b/rul_adapt/approach/pseudo_labels.py
@@ -153,6 +153,10 @@ def __init__(
         self._inductive = inductive
         self._max_rul = max_rul
 
+    @property
+    def dataset_name(self) -> str:
+        return self._reader.dataset_name
+
     @property
     def _patched_split(self) -> str:
         return "test" if self._inductive else "dev"