fix hydra problem

app.py

@@ -70,21 +70,12 @@ class DemoRunner:
         cfg['target_task_name'] = instruction
 
         # self._env.cache_video = []
-        # self._md_logger.clear()
+        self.simulation_runner._md_logger = ''
 
-        try:
-            self.simulation_runner.task_creation()
-            self.simulation_runner.simulate_task()
-        except Exception as e:
-            return f'Error: {e}', None, None
-
-        # video_file_name = None
-        # if self._env.cache_video:
-        #     rendered_clip = ImageSequenceClip(self._env.cache_video, fps=25)
-        #     video_file_name = NamedTemporaryFile(suffix='.mp4').name
-        #     rendered_clip.write_videofile(video_file_name, fps=25)
-        info = '### Run'
-        return info, self.simulation_runner.video_path
+        self.simulation_runner.task_creation()
+        self.simulation_runner.simulate_task()
+        print("self.video_path = ", self.simulation_runner.video_path)
+        return self.simulation_runner._md_logger, self.simulation_runner.video_path
 
 
 def setup(api_key):
@@ -99,7 +90,9 @@ def setup(api_key):
 
 def run(instruction, demo_runner):
     if demo_runner is None:
-        return 'Please run setup first!', None, None
+        return 'Please run setup first!', None
+
+    # return None, "/home/baochen/Desktop/projects/GenSim2/data/assemble-pallet-ball-train/videos/000001.mp4"
     return demo_runner.run(instruction)
 
 

cliport/cfg/config.yaml

@@ -5,7 +5,7 @@ tag: default
 debug: False
 gpt_temperature: 0.8 # GPT-4 response temperature. higher means more diversity
 prompt_folder: vanilla_task_generation_prompt # the prompt folder that stores the prompt chain
-max_env_run_cnt: 3 # maximum number of runs for each environment
+max_env_run_cnt: 1 # maximum number of runs for each environment
 trials: 10 # how many times of spawning each environment generated
 output_folder: 'output/output_stats'
 model_output_dir: '' # to be filled in with date

gensim/sim_runner.py

@@ -50,7 +50,8 @@ class SimulationRunner:
         self.generated_tasks = []
         self.passed_tasks = []  # accepted ones
 
-        self.video_path = ""
+        self.video_path = None
+        self._md_logger = ''
 
     def print_current_stats(self):
         """ print the current statistics of the simulation design """
@@ -196,6 +197,7 @@ class SimulationRunner:
             save_text(self.cfg['model_output_dir'], self.generated_task_name + '_error', str(traceback.format_exc()))
             print("========================================================")
             print("Syntax Exception:", to_print)
+            self._md_logger = str(traceback.format_exc())
             return
 
         try:
@@ -218,4 +220,6 @@ class SimulationRunner:
             save_text(self.cfg['model_output_dir'], self.generated_task_name + '_error', str(traceback.format_exc()))
             print("========================================================")
             print("Runtime Exception:", to_print)
+            self._md_logger = str(traceback.format_exc())
+
         self.memory.save_run(self.generated_task)