suppress errors in `fbcode/vision` - batch 2

Differential Revision: D30222339 fbshipit-source-id: 97d498df72ef897b8dc2405764e3ffd432082e3c

suppress errors in `fbcode/vision` - batch 2
Differential Revision: D30222339 fbshipit-source-id: 97d498df72ef897b8dc2405764e3ffd432082e3c
9db70400 · Pyre Bot Jr · Facebook GitHub Bot · 80411783 · 9db70400
Commit 9db70400 authored Aug 10, 2021 by Pyre Bot Jr Committed by Facebook GitHub Bot Aug 10, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

pytorch3d/renderer/implicit/raysampling.py pytorch3d/renderer/implicit/raysampling.py +1 -1

No files found.
--- a/pytorch3d/renderer/implicit/raysampling.py
+++ b/pytorch3d/renderer/implicit/raysampling.py
@@ -123,7 +123,7 @@ class GridRaysampler(torch.nn.Module):
        device = cameras.device

        # expand the (H, W, 2) grid batch_size-times to (B, H, W, 2)
-        xy_grid = self._xy_grid.to(device)[None].expand(  # pyre-ignore
+        xy_grid = self._xy_grid.to(device)[None].expand(
            batch_size, *self._xy_grid.shape
        )