Update two_arm_octopus_ai_imp.py
@@ -172,9 +172,11 @@ def main():
user_defined_condition = user_defined_condition_function(reward, systems, time)
if user_defined_condition == True:
print(" User defined condition satisfied, exit simulation")
- done = True
+ print(" Episode finished after {} ".format(time))
+ break
- # If done=True, Exit the simulation loop before, reaching final time
+ # If done=True, NaN detected in simulation.
+ # Exit the simulation loop before, reaching final time
if done:
print(" Episode finished after {} ".format(time))
break
