Skip to content

Commit c347ff3

Browse files
committed
Add prompt template with grid explanation
1 parent 6e49d40 commit c347ff3

File tree

2 files changed

+22
-1
lines changed

2 files changed

+22
-1
lines changed
Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,21 @@
1+
You are an intelligent assistant who is very good at answering test questions accurately.
2+
In the examples that follow you will be shown grids of numbers.
3+
The numbers in the grids range from 0 through 9.
4+
Each grid can be rendered as a grid of squares.
5+
Each square in the grid is rendered as a colored square where the color of the square is derived from the number.
6+
The colors are decided as follows:
7+
8+
0 - black
9+
1 - blue
10+
2 - red
11+
3 - green
12+
4 - yellow
13+
5 - grey
14+
6 - magenta
15+
7 - brown
16+
8 - cyan
17+
9 - maroon
18+
19+
With that in mind, do your best to solve the question below.
20+
21+
{{ prompt }}

eureka_ml_insights/user_configs/arc_agi.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ def configure_pipeline(self, model_config=None, resume_from=None, resume_logdir=
5959
self.data_processing_comp = PromptProcessingConfig(
6060
component_type=PromptProcessing,
6161
prompt_template_path=os.path.join(
62-
os.path.dirname(__file__), "../prompt_templates/arc_agi_templates/arc_agi_v1_basic.jinja"
62+
os.path.dirname(__file__), "../prompt_templates/arc_agi_templates/arc_agi_v1_grid_explanation.jinja"
6363
),
6464
data_reader_config=DataSetConfig(
6565
HFDataReader,

0 commit comments

Comments
 (0)