Commit 766429e8 authored by Matt Tancik's avatar Matt Tancik
Browse files

Fix attribute bug

parent 7995c54a
......@@ -57,7 +57,7 @@ class OccupancyField(nn.Module):
grid_coords: The grid coordinates. It is a tensor of shape (num_cells, num_dim).
grid_indices: The grid indices. It is a tensor of shape (num_cells,).
"""
aabb = torch.Tensor
aabb: torch.Tensor
occ_grid: torch.Tensor
occ_grid_binary: torch.Tensor
grid_coords: torch.Tensor
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment