tests: adjusted tests to changes

Author: jmatejcz

tests/rai_bench/tool_calling_agent/test_predefined_basic_tasks.py

@@ -20,6 +20,18 @@
     TaskArgs,
 )
 from rai_bench.tool_calling_agent.predefined.basic_tasks import (
+    BOX1_ENTITY,
+    BOX1_POSITION,
+    BOX2_ENTITY,
+    BOX2_POSITION,
+    DEFAULT_DINO_CONFIDENCE,
+    DEFAULT_FPS,
+    DEFAULT_PUBLISH_FREQUENCY,
+    DEFAULT_SAM_CONFIDENCE,
+    DINO_CONFIDENCE_2,
+    FPS_2,
+    SAM_CONFIDENCE_2,
+    TOMATO_ENTITY,
     all_camera_images_notord_val,
     check_spawnable_entities_val,
     color_image_ord_val,
@@ -31,20 +43,10 @@
     topics_ord_val,
 )
 from rai_bench.tool_calling_agent.tasks.basic import (
-    BOX1_ENTITY,
-    BOX1_POSITION,
-    BOX2_ENTITY,
-    BOX2_POSITION,
     COLOR_IMAGE_TOPIC,
-    DEFAULT_DINO_CONFIDENCE,
-    DEFAULT_FPS,
-    DEFAULT_PUBLISH_FREQUENCY,
-    DEFAULT_SAM_CONFIDENCE,
     DELETE_ENTITY_SERVICE,
     DELETE_ENTITY_TYPE,
     DEPTH_IMAGE_TOPIC,
-    DINO_CONFIDENCE_2,
-    FPS_2,
     GET_PARAMETERS_TYPE,
     GET_SPAWNABLE_NAMES_SERVICE,
     GET_WORLD_PROPERTIES_TYPE,
@@ -59,12 +61,10 @@
     ROBOT_STATE_PUBLISHER_GET_PARAMS,
     ROBOT_STATE_PUBLISHER_LIST_PARAMS,
     ROBOT_STATE_PUBLISHER_SET_PARAMS,
-    SAM_CONFIDENCE_2,
     SET_PARAMETERS_ATOMICALLY_TYPE,
     SET_PARAMETERS_TYPE,
     SPAWN_ENTITY_SERVICE,
     SPAWN_ENTITY_TYPE,
-    TOMATO_ENTITY,
     CheckSpawnableEntitiesTask,
     ConfigureVisionPipelineTask,
     GetAllROS2CamerasTask,