def _after_step()

in gym-compete/gym_compete/new_envs/multi_monitoring.py [0:0]

• 8 lines of code
• 4 McCabe index (conditional complexity)

    def _after_step(self, observation, reward, done, info):
        if not self.enabled: return done

        if done[0] and self.env_semantics_autoreset:
            # For envs with BlockingReset wrapping VNCEnv, this observation will be the first one of the new episode
            self._reset_video_recorder()
            self.episode_id += 1
            self._flush()

        # Record video
        self.video_recorder.capture_frame()

        return done