Method finalize

4-atari-hard/env.py:115–128

Final 'latest' checkpoint + a final.json result summary.

(self, frames, game_returns, state_fn, k=100)

            _atomic_save(state_fn(), os.path.join(self.ckpt_dir, "best.pt"))

    def finalize(self, frames, game_returns, state_fn, k=100):
        """Final 'latest' checkpoint + a final.json result summary."""
        if self.ckpt_dir:
            _atomic_save(state_fn(), os.path.join(self.ckpt_dir, "latest.pt"))
        if self.dir:
            tail = [float(x) for x in game_returns[-k:]]
            with open(os.path.join(self.dir, "final.json"), "w") as fh:
                json.dump({"frames_total": frames, "frames_unit": "agent_steps",
                           "gate_metric": "game_return_mean_lastK", "K": k,
                           "value_mean": statistics.fmean(tail) if tail else float("nan"),
                           "value_std": statistics.pstdev(tail) if len(tail) > 1 else 0.0,
                           "episodes_counted": len(tail)}, fh, indent=1)
        if self.f:
            self.f.close()


# Gymnasium / ALE id (used by make_env / --test rendering) paired with the