From b6233dff3cec42696f2ea0eea286ded48f02e79b Mon Sep 17 00:00:00 2001 From: Eric Liang Date: Sat, 15 Feb 2020 17:22:23 -0800 Subject: [PATCH] [rllib] Fix bad sample count assert --- rllib/optimizers/rollout.py | 1 - 1 file changed, 1 deletion(-) diff --git a/rllib/optimizers/rollout.py b/rllib/optimizers/rollout.py index cc0a80a35..4d8611f2d 100644 --- a/rllib/optimizers/rollout.py +++ b/rllib/optimizers/rollout.py @@ -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)