def _write_and_reset_trajectory()

in gym3/trajectory_recorder.py [0:0]

• 9 lines of code
• 1 McCabe index (conditional complexity)

    def _write_and_reset_trajectory(self, idx) -> None:
        filepath = os.path.join(
            self._directory, f"{self._prefix}{self._episode_count:09d}.pickle"
        )
        with open(filepath, "wb") as f:
            self._trajectories[idx]['reward'] = np.array(self._trajectories[idx]['reward'])
            pickle.dump(self._trajectories[idx], f)
        self._trajectories[idx] = self._new_trajectory_dict()
        self._episode_count += 1