fix example

juaml · Jul 30, 2024 · 6add24f · 6add24f
1 parent 1979058
commit 6add24f
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/examples/03_complex_models/run_generate_target.py b/examples/03_complex_models/run_generate_target.py
@@ -57,11 +57,9 @@
 # be the the `generate_target`, applying to the "petal" features and using the
 # target_creator pipeline as the transformer.
 creator = PipelineCreator(problem_type="regression")
-creator.add("zscore")
+creator.add("zscore", apply_to="*")
 creator.add("generate_target", apply_to="petal", transformer=target_creator)
-creator.add(
-    "linreg", apply_to="sepal",
-)
+creator.add("linreg", apply_to="sepal")
 
 ###############################################################################
 # We finally evaluate the model within the cross validation.