From 24ca65d7390a5d00f3688dfe28d027e610b2b4f1 Mon Sep 17 00:00:00 2001
From: jkaminski <jkaminski@niuitmo.ru>
Date: Mon, 8 Jul 2024 18:51:34 +0300
Subject: [PATCH] attribute redirection

---
 bamt/core/node_models/classifier.py | 7 +++++++
 bamt/core/node_models/regressor.py  | 7 +++++++
 2 files changed, 14 insertions(+)

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