From 46a8d4873d8eac08d978e53c2062caa1ee2c773a Mon Sep 17 00:00:00 2001 From: kywch Date: Wed, 27 Mar 2024 15:17:13 -0700 Subject: [PATCH] error fix --- analysis/proc_task_cond_result.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/analysis/proc_task_cond_result.py b/analysis/proc_task_cond_result.py index 140d7677..53b10468 100644 --- a/analysis/proc_task_cond_result.py +++ b/analysis/proc_task_cond_result.py @@ -121,11 +121,9 @@ def get_event_stats(task_name, task_data): return results if __name__ == '__main__': - # parser = argparse.ArgumentParser(description='Process replay data') - # parser.add_argument('policy_store_dir', type=str, help='Path to the policy directory') - # args = parser.parse_args() - - args = argparse.Namespace(policy_store_dir='pol_task_cond2') + parser = argparse.ArgumentParser(description='Process replay data') + parser.add_argument('policy_store_dir', type=str, help='Path to the policy directory') + args = parser.parse_args() # Gather the event data by tasks, across multiple replays data_by_task = gather_agent_events_by_task(args.policy_store_dir)