fix cuml attrs

tvdboom · Feb 1, 2024 · 7cef876 · 7cef876
1 parent 4cb4637
commit 7cef876
Show file tree

Hide file tree

Showing 4 changed files with 10,181 additions and 2 deletions.
diff --git a/atom/basetransformer.py b/atom/basetransformer.py
@@ -457,9 +457,11 @@ def _get_est_class(self, name: str, module: str) -> type[Estimator]:
 
         """
         try:
-            return getattr(import_module(f"{self.engine.estimator}.{module}"), name)
+            mod = import_module(f"{self.engine.estimator}.{module}")
         except (ModuleNotFoundError, AttributeError):
-            return getattr(import_module(f"sklearn.{module}"), name)
+            mod = import_module(f"sklearn.{module}")
+
+        return self._inherit(getattr(mod, name))
 
     @staticmethod
     @overload

diff --git a/docs_sources/examples/univariate_forecast.ipynb b/docs_sources/examples/univariate_forecast.ipynb
diff --git a/examples/univariate_forecast.ipynb b/examples/univariate_forecast.ipynb
diff --git a/mkdocs.yml b/mkdocs.yml
@@ -315,6 +315,7 @@ nav:
         - Regression: examples/regression.ipynb
         - Successive halving: examples/successive_halving.ipynb
         - Train sizing: examples/train_sizing.ipynb
+        - Univariate forecast: examples/univariate_forecast.ipynb
         - Utilities: examples/utilities.ipynb
     - Changelog:
         - v6.x.x: changelog/v6.x.x.md