attribute redirection

jrzkaminski · Jul 8, 2024 · 24ca65d · 24ca65d
1 parent 6768931
commit 24ca65d
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/bamt/core/node_models/classifier.py b/bamt/core/node_models/classifier.py
@@ -23,3 +23,10 @@ def predict_proba(self, X: np.ndarray) -> np.ndarray:
 
     def __str__(self):
         return str(self._classifier)
+
+    def __getattr__(self, name: str):
+        if self._classifier:
+            return getattr(self._classifier, name)
+        raise AttributeError(
+            f"'{self.__class__.__name__}' object has no attribute '{name}'"
+        )
diff --git a/bamt/core/node_models/regressor.py b/bamt/core/node_models/regressor.py
@@ -23,3 +23,10 @@ def predict_proba(self, X: np.ndarray) -> np.ndarray:
 
     def __str__(self):
         return str(self._regressor)
+
+    def __getattr__(self, name: str):
+        if self._regressor:
+            return getattr(self._regressor, name)
+        raise AttributeError(
+            f"'{self.__class__.__name__}' object has no attribute '{name}'"
+        )