Fix bug in structural variables names during bootstrap

Labo-Lacourse · Aug 15, 2023 · 9424f8f · 9424f8f
1 parent 855cadd
commit 9424f8f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/stepmix/bootstrap.py b/stepmix/bootstrap.py
@@ -73,7 +73,7 @@ def bootstrap(
     check_is_fitted(estimator)
     n_samples = X.shape[0]
     x_names = estimator.x_names_
-    y_names = estimator.y_names_ if hasattr(estimator, "y_names") else None
+    y_names = estimator.y_names_ if hasattr(estimator, "y_names_") else None
 
     # Use the estimator built-in method to check the input
     # This will ensure that X and Y are numpy arrays for the rest of the bootstrap procedure