accomodate both versions of vr demo

StanfordVL · Mar 24, 2024 · 438d779 · 438d779
1 parent b6e9550
commit 438d779
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/igibson/examples/behavior/behavior_demo_replay.py b/igibson/examples/behavior/behavior_demo_replay.py
@@ -128,8 +128,12 @@ def replay_demo(
     vr_settings = VrSettings(config_str=IGLogReader.read_metadata_attr(in_log_path, "/metadata/vr_settings"))
     vr_settings.set_frame_save_path(frame_save_path)
 
-    task = IGLogReader.read_metadata_attr(in_log_path, "/metadata/task_name")
-    task_id = IGLogReader.read_metadata_attr(in_log_path, "/metadata/task_instance")
+    task = IGLogReader.read_metadata_attr(in_log_path, "/metadata/atus_activity")
+    if task is None:
+        task = IGLogReader.read_metadata_attr(in_log_path, "/metadata/task_name")
+    task_id = IGLogReader.read_metadata_attr(in_log_path, "/metadata/activity_definition")
+    if task_id is None:
+        task_id = IGLogReader.read_metadata_attr(in_log_path, "/metadata/task_instance")
     scene = IGLogReader.read_metadata_attr(in_log_path, "/metadata/scene_id")
     physics_timestep = IGLogReader.read_metadata_attr(in_log_path, "/metadata/physics_timestep")
     render_timestep = IGLogReader.read_metadata_attr(in_log_path, "/metadata/render_timestep")