fix

sim/simulator.py

@@ -226,8 +226,6 @@ class GenieSimulator(LearnedSimulator):
 
 
     def set_initial_state(self, state: Tuple[np.ndarray, np.ndarray]):
-        if not self.allow_external_prompt and self.gt_phys_sim is not None:
-            raise NotImplementedError("Initial state is set by the physics simulator.")
         self.init_prompt = state