Update gensim/sim_runner.py

gensim/sim_runner.py

@@ -60,7 +60,7 @@ class SimulationRunner:
         """ save the final simulation statistics """
         self.diversity_score = compute_diversity_score_from_assets(self.task_asset_logs, self.curr_trials)
         save_stat(self.cfg, self.cfg['model_output_dir'], self.generated_tasks, self.syntax_pass_rate / (self.curr_trials),
-                self.runtime_pass_rate / (self.curr_trials), self.env_pass_rate / (self.curr_trials), self.diversity_score)
+                self.runtime_pass_rate / (self.data_pathcurr_trials), self.env_pass_rate / (self.curr_trials), self.diversity_score)
         print("Model Folder: ", self.cfg['model_output_dir'])
         print(f"Total {len(self.generated_tasks)} New Tasks:", [task['task-name'] for task in self.generated_tasks])
         try:
@@ -92,7 +92,7 @@ class SimulationRunner:
                     online_code_buffer[task_file] = open("cliport/generated_tasks/" + task_file).read()
 
             random_task_file = random.sample(list(online_code_buffer.keys()), 1)[0]
-            class_def = [line for line in online_code_buffer[task_file].split("\n") if line.startswith('class')]
+            class_def = [line for line in online_code_buffer[random_task_file].split("\n") if line.startswith('class')]
             task_name = class_def[0]
             task_name = task_name[task_name.find("class "): task_name.rfind("(Task)")][6:]
             self.curr_task_name = self.generated_task_name = task_name