agent_training.py

class GymTask(Task):
 
    # ...
  
    def execute(self, *args: Any, **kwargs: Any) -> None:
        if not self._proxy_env.is_rl_agent_trained and not self.is_rl_agent_training:
            self._start_training()
        if self._proxy_env.is_rl_agent_trained and self.is_rl_agent_training:
            self._stop_training()
            
    # ...
    
    def _start_training(self) -> None:
        self.is_rl_agent_training = True
        self._fit(self._proxy_env, self.nb_steps)

    def _stop_training(self) -> None:
        if self.is_rl_agent_training:
            self.is_rl_agent_training = False
            self._proxy_env.close()