2.0.0 dev (#121)

* requirements update * minor change * minor changes * initial index commit * attribute redirection * minor change * blank line
aimclub · Jul 8, 2024 · 638b792 · 638b792
1 parent d89e91d
commit 638b792
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}'"
+        )