Merge pull request #38 from AutoResearch/37-bug-get_models-crashese

Update regressor.py
AutoResearch · Aug 23, 2023 · 643343a · 643343a
2 parents 27df0d0 + 9d757a8
commit 643343a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/autora/theorist/bms/regressor.py b/src/autora/theorist/bms/regressor.py
@@ -194,7 +194,7 @@ def get_models(self):
             bms_model = BMSRegressor()
             bms_model.model_ = tree
             bms_model.temp_ = self.ts[idx]
-            bms_model.variables = list(X.columns) if hasattr(X, "columns") else ["X%d" % i for i in range(X.shape[1])]
+            bms_model.variables = list(self.X_.columns) if hasattr(self.X_, "columns") else ["X%d" % i for i in range(self.X_.shape[1])]
             model_list.append(bms_model)
         return model_list