[rllib] Fix bad sample count assert

2026-06-27 19:16:19 +08:00 · 2020-02-15 17:22:23 -08:00
parent 2e60f0d4d8
commit b6233dff3c
1 changed files with 0 additions and 1 deletions
@@ -23,7 +23,6 @@ def collect_samples(agents, sample_batch_size, num_envs_per_worker,
        [fut_sample], _ = ray.wait(list(agent_dict))
        agent = agent_dict.pop(fut_sample)
        next_sample = ray_get_and_free(fut_sample)
-        assert next_sample.count >= sample_batch_size * num_envs_per_worker
        num_timesteps_so_far += next_sample.count
        trajectories.append(next_sample)