solve dependencies problem

task_gen_dependencies/task_generate.py

@@ -99,6 +99,7 @@ class OldTaskGenerator:
     def __init__(self, task_template, data_root):
         self.data_root = data_root
         self.init_info(task_template)
+        self.task_template = task_template
 
     def _load_json(self, relative_path):
         with open(os.path.join(self.data_root, relative_path), 'r') as file:
@@ -224,10 +225,11 @@ class OldTaskGenerator:
 
     def generate_tasks(self, save_path, task_num, task_name):
         os.makedirs(save_path, exist_ok=True)
-
+        generated_tasks_path = []
         for i in range(task_num):
             task_instance = deepcopy(self.task_template)
             output_file = os.path.join(save_path, f'{task_name}_%d.json' % (i))
+            generated_tasks_path.append(output_file)
             task_instance['objects'] = []
             task_instance['objects'] += self.fix_obj_infos
             task_instance['robot'] = {
@@ -236,6 +238,7 @@ class OldTaskGenerator:
                 "robot_cfg": self.robot_cfg,
                 "stand": self.stand,
             }
+            task_instance['recording_setting'] = self.task_template["recording_setting"]
 
             flag_failed = False
             for key in self.layouts:
@@ -258,5 +261,5 @@ class OldTaskGenerator:
             Log.info('Saved task json to %s' % output_file)
             with open(output_file, 'w') as f:
                 json.dump(task_instance, f, indent=4)
-
+        return generated_tasks_path
 # -------------------------------------------------------------