fix a bug in feature selection prompt (#333)

microsoft · Sep 25, 2024 · 4cdbf0e · 4cdbf0e
1 parent 3d36c45
commit 4cdbf0e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/rdagent/scenarios/kaggle/prompts.yaml b/rdagent/scenarios/kaggle/prompts.yaml
@@ -272,9 +272,9 @@ feature_selection_feedback_generation:
     Result: {{exp.result}}
 
     Available Features:
-    {% for feature in available_features %}
-    - {{feature.name}}: {{feature.description}}
-      Shape: {{feature.shape}}
+    {% for feature, shape in available_features %}
+      Features description: {{feature}}
+      Feature shape: {{shape}}
     {% endfor %}
 
     Compare and observe the results. Which result has a better return and lower risk? If the performance increases, the hypothesis should be considered positive (working).