fix: negative candidate ids (#696)

emma-heriot-watt · Feb 6, 2023 · 8a27cf6 · 8a27cf6
1 parent 564a2b0
commit 8a27cf6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/emma_policy/datamodules/simbot_action_dataset.py b/src/emma_policy/datamodules/simbot_action_dataset.py
@@ -6,6 +6,7 @@
 from emma_datasets.constants.simbot.simbot import get_arena_definitions
 from emma_datasets.datamodels.datasets.utils.simbot_utils.instruction_processing import (
     get_object_label_from_object_id,
+    get_object_readable_name_from_object_id,
 )
 from emma_datasets.datamodels.datasets.utils.simbot_utils.paraphrasers import (
     InstructionParaphraser,
@@ -197,10 +198,12 @@ def simbot_vision_augmentation(  # noqa: WPS210, WPS231
             )
             source_text = format_instruction(source_text)
 
-            object_name = get_object_label_from_object_id(
+            object_name = get_object_readable_name_from_object_id(
                 object_id=action_object_metadata["id"][object_candidate_idx],
                 object_assets_to_names=self._object_assets_to_names,
+                special_name_cases=self._special_name_cases,
             )
+
             object_token = None
 
             # We need to skip the instances that are from annotations aka paraphrasable