clean-up
Browse files- 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()
|