From 56f75d30212d7439fe0d81e3ca71d1d88f1569c0 Mon Sep 17 00:00:00 2001
From: perib <pedroribeiroh@gmail.com>
Date: Wed, 6 Sep 2023 14:14:25 -0700
Subject: [PATCH] estimator type

---
 tpot/base.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tpot/base.py b/tpot/base.py
index c1e3a3fb..676038f8 100644
--- a/tpot/base.py
+++ b/tpot/base.py
@@ -2031,6 +2031,9 @@ def classes_(self):
         """The classes labels. Only exist if the last step is a classifier."""
         return self.fitted_pipeline_.classes_
 
+    @property
+    def _estimator_type(self):
+        return self.fitted_pipeline_._estimator_type
 
 def _has_cuml():
     try: