Skip to content

Commit c65d470

Browse files
committed
refactor:: removed redundant code
1 parent 4a03c30 commit c65d470

File tree

1 file changed

+4
-64
lines changed

1 file changed

+4
-64
lines changed

src/rai_bench/rai_bench/tool_calling_agent/predefined/spatial_reasoning_tasks.py

Lines changed: 4 additions & 64 deletions
Original file line numberDiff line numberDiff line change
@@ -32,66 +32,6 @@
3232
)
3333

3434
IMG_PATH = "src/rai_bench/rai_bench/tool_calling_agent/predefined/images/"
35-
true_response_inputs: List[BoolImageTaskInput] = [
36-
BoolImageTaskInput(
37-
question="Is the door on the left from the desk?",
38-
images_paths=[IMG_PATH + "image_1.jpg"],
39-
),
40-
BoolImageTaskInput(
41-
question="Is the light on in the room?",
42-
images_paths=[IMG_PATH + "image_2.jpg"],
43-
),
44-
BoolImageTaskInput(
45-
question="Do you see the plant?",
46-
images_paths=[IMG_PATH + "image_2.jpg"],
47-
),
48-
BoolImageTaskInput(
49-
question="Are there any pictures on the wall?",
50-
images_paths=[IMG_PATH + "image_3.jpg"],
51-
),
52-
BoolImageTaskInput(
53-
question="Are there 3 pictures on the wall?",
54-
images_paths=[IMG_PATH + "image_4.jpg"],
55-
),
56-
BoolImageTaskInput(
57-
question="Is there a plant behind the rack?",
58-
images_paths=[IMG_PATH + "image_5.jpg"],
59-
),
60-
BoolImageTaskInput(
61-
question="Is there a pillow on the armchain?",
62-
images_paths=[IMG_PATH + "image_7.jpg"],
63-
),
64-
]
65-
false_response_inputs: List[BoolImageTaskInput] = [
66-
BoolImageTaskInput(
67-
question="Is the door open?",
68-
images_paths=[IMG_PATH + "image_1.jpg"],
69-
),
70-
BoolImageTaskInput(
71-
question="Is someone in the room?",
72-
images_paths=[IMG_PATH + "image_1.jpg"],
73-
),
74-
BoolImageTaskInput(
75-
question="Do you see the plant?",
76-
images_paths=[IMG_PATH + "image_3.jpg"],
77-
),
78-
BoolImageTaskInput(
79-
question="Are there 4 pictures on the wall?",
80-
images_paths=[IMG_PATH + "image_4.jpg"],
81-
),
82-
BoolImageTaskInput(
83-
question="Is there a rack on the left from the sofa?",
84-
images_paths=[IMG_PATH + "image_4.jpg"],
85-
),
86-
BoolImageTaskInput(
87-
question="Is there a plant on the right from the window?",
88-
images_paths=[IMG_PATH + "image_6.jpg"],
89-
),
90-
BoolImageTaskInput(
91-
question="Is there a red pillow on the armchair?",
92-
images_paths=[IMG_PATH + "image_7.jpg"],
93-
),
94-
]
9535
########## SUBTASKS #################################################################
9636
return_true_subtask = CheckArgsToolCallSubTask(
9737
expected_tool_name="return_bool_response", expected_args={"response": True}
@@ -141,10 +81,6 @@ def get_spatial_tasks(
14181
question="is there a TV in the room?",
14282
images_paths=[IMG_PATH + "image_4.jpg"],
14383
),
144-
BoolImageTaskInput(
145-
question="Is there a pillow on the armchain?",
146-
images_paths=[IMG_PATH + "image_7.jpg"],
147-
),
14884
]
14985

15086
medium_true_inputs = [
@@ -181,6 +117,10 @@ def get_spatial_tasks(
181117
question="Is there a rug under the bed?",
182118
images_paths=[IMG_PATH + "image_2.jpg"],
183119
),
120+
BoolImageTaskInput(
121+
question="Is there a pillow on the armchain?",
122+
images_paths=[IMG_PATH + "image_7.jpg"],
123+
),
184124
]
185125

186126
easy_false_inputs = [

0 commit comments

Comments
 (0)