diff --git a/bamt/core/node_models/classifier.py b/bamt/core/node_models/classifier.py index afe9df4..f68e478 100644 --- 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 index d69941a..a18a95c 100644 --- 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}'" + ) \ No newline at end of file