clear base class cache from worldvoxelcollision

This commit is contained in:
Balakumar Sundaralingam
2024-04-26 14:36:43 -07:00
parent 3f1beb991f
commit 1e0b5a80c2

View File

@@ -704,6 +704,7 @@ class WorldVoxelCollision(WorldMeshCollision):
- self.max_esdf_distance - self.max_esdf_distance
).to(dtype=self._voxel_tensor_list[3].dtype) ).to(dtype=self._voxel_tensor_list[3].dtype)
self._env_n_voxels[:] = 0 self._env_n_voxels[:] = 0
super().clear_cache()
def get_voxel_grid_shape(self, env_idx: int = 0, obs_idx: int = 0): def get_voxel_grid_shape(self, env_idx: int = 0, obs_idx: int = 0):
return self._voxel_tensor_list[3][env_idx, obs_idx].shape return self._voxel_tensor_list[3][env_idx, obs_idx].shape