diff --git a/igibson/examples/behavior/behavior_demo_replay.py b/igibson/examples/behavior/behavior_demo_replay.py index f731e6154..c8548fef4 100644 --- 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")