set resize_size in wrapper

octo-models · Jan 8, 2024 · 756ae9a · 756ae9a
1 parent 8576118
commit 756ae9a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/octo/utils/gym_wrappers.py b/octo/utils/gym_wrappers.py
@@ -222,10 +222,11 @@ def __init__(
             self.observation_space, gym.spaces.Dict
         ), "Only Dict observation spaces are supported."
         spaces = self.observation_space.spaces
+        self.resize_size = resize_size
 
         if resize_size is None:
             self.keys_to_resize = {}
-        elif isinstance(self.resize_size, tuple):
+        elif isinstance(resize_size, tuple):
             self.keys_to_resize = {k: resize_size for k in spaces if "image_" in k}
         else:
             self.keys_to_resize = {