off-by-one bug in % hyperparameter

kzl · Feb 7, 2022 · e2d82e6 · e2d82e6
1 parent f04280e
commit e2d82e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gym/experiment.py b/gym/experiment.py
@@ -106,7 +106,7 @@ def experiment(
     num_trajectories = 1
     timesteps = traj_lens[sorted_inds[-1]]
     ind = len(trajectories) - 2
-    while ind >= 0 and timesteps + traj_lens[sorted_inds[ind]] < num_timesteps:
+    while ind >= 0 and timesteps + traj_lens[sorted_inds[ind]] <= num_timesteps:
         timesteps += traj_lens[sorted_inds[ind]]
         num_trajectories += 1
         ind -= 1