nvkartik commited on
Commit
d51ca30
·
1 Parent(s): f0f6cdb
Files changed (1) hide show
  1. isaaclab_env_wrapper.py +0 -11
isaaclab_env_wrapper.py CHANGED
@@ -177,19 +177,8 @@ class IsaacLabEnvWrapper(gym.vector.AsyncVectorEnv):
177
  placeholder = np.zeros((480, 640, 3), dtype=np.uint8)
178
  return [placeholder] * self._num_envs
179
 
180
- # if frames.ndim == 4:
181
- print(f"Returning frames")
182
- print(f"Frames type: {type(frames)}")
183
- print(f"Frames shape: {frames.shape}")
184
- print(f"Frames dtype: {frames.dtype}")
185
- # [frames[i] for i in range(min(len(frames), self._num_envs))]
186
  return [frames] * self._num_envs
187
 
188
- # print(f"Returnig zero frames")
189
- # print(f"Frames shape: {frames.shape}")
190
-
191
- # return [np.zeros((480, 640, 3), dtype=np.uint8)] * self._num_envs
192
-
193
  def render(self) -> np.ndarray | None:
194
  """Render all environments and return list of frames."""
195
  self._check_closed()
 
177
  placeholder = np.zeros((480, 640, 3), dtype=np.uint8)
178
  return [placeholder] * self._num_envs
179
 
 
 
 
 
 
 
180
  return [frames] * self._num_envs
181
 
 
 
 
 
 
182
  def render(self) -> np.ndarray | None:
183
  """Render all environments and return list of frames."""
184
  self._check_closed()