Commit 1836c786 authored by Pyre Bot Jr's avatar Pyre Bot Jr Committed by Facebook GitHub Bot
Browse files

suppress errors in `vision/fair/pytorch3d`

Differential Revision: D32310496

fbshipit-source-id: fa1809bbe0b999baee6d07fad3890dc8c2a2157b
parent cac6cb1b
...@@ -493,7 +493,7 @@ class Meshes: ...@@ -493,7 +493,7 @@ class Meshes:
# NOTE consider converting index to cpu for efficiency # NOTE consider converting index to cpu for efficiency
if index.dtype == torch.bool: if index.dtype == torch.bool:
# advanced indexing on a single dimension # advanced indexing on a single dimension
index = index.nonzero() # pyre-ignore index = index.nonzero()
index = index.squeeze(1) if index.numel() > 0 else index index = index.squeeze(1) if index.numel() > 0 else index
index = index.tolist() index = index.tolist()
verts = [self.verts_list()[i] for i in index] verts = [self.verts_list()[i] for i in index]
......
...@@ -351,7 +351,7 @@ class Pointclouds: ...@@ -351,7 +351,7 @@ class Pointclouds:
# NOTE consider converting index to cpu for efficiency # NOTE consider converting index to cpu for efficiency
if index.dtype == torch.bool: if index.dtype == torch.bool:
# advanced indexing on a single dimension # advanced indexing on a single dimension
index = index.nonzero() # pyre-ignore index = index.nonzero()
index = index.squeeze(1) if index.numel() > 0 else index index = index.squeeze(1) if index.numel() > 0 else index
index = index.tolist() index = index.tolist()
points = [self.points_list()[i] for i in index] points = [self.points_list()[i] for i in index]
......
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