changed output of single round

ControllerExecutorFlow.py

@@ -87,14 +87,14 @@ class ControllerExecutorFlow(CompositeFlow):
                 "observation":  self.ask_subflow(controller_reply["command"], controller_reply["command_args"]).get_data()
         }
             
-        return executor_reply
+        return controller_reply, executor_reply
     
     def run(self,input_data):
 
         executor_reply = input_data
         
         for round in range(self.flow_config["max_rounds"]):
-            executor_reply = self._single_round_controller_executor(executor_reply)
+            _ , executor_reply = self._single_round_controller_executor(executor_reply)
             
             if executor_reply.get("EARLY_EXIT",False):
                 return executor_reply