Merge pull request octo-models#73 from moojink/main

Fix pad_mask bug: min instead of max for pad_length
wjj2024 · Apr 9, 2024 · bd930f9 · bd930f9
2 parents 8559a70 + 8b0e267
commit bd930f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/octo/utils/gym_wrappers.py b/octo/utils/gym_wrappers.py
@@ -17,7 +17,7 @@ def stack_and_pad(history: list, num_obs: int):
     """
     horizon = len(history)
     full_obs = {k: np.stack([dic[k] for dic in history]) for k in history[0]}
-    pad_length = horizon - max(num_obs, horizon)
+    pad_length = horizon - min(num_obs, horizon)
     pad_mask = np.ones(horizon)
     pad_mask[:pad_length] = 0
     full_obs["pad_mask"] = pad_mask