Method assign

4-atari-hard/env_robustify.py:198–202

Stagger envs across a window below max_starting_point.

(self, envs)

        self.success = np.zeros(self.n + 1, dtype=np.float64)

    def assign(self, envs):
        """Stagger envs across a window below max_starting_point."""
        per = max(self.window // max(self.n_envs, 1), 1)
        for i, e in enumerate(envs):
            e.starting_point = max(self.max_starting_point - i * per, 0)

    def record(self, starting_point, success):
        # exponential-ish freshening: latest wins (atari-reset keeps last rate)