Add missing dtype parameter type annotations

Summary: Add missing dtype parameter type annotations

Reviewed By: nikhilaravi

Differential Revision: D28943370

fbshipit-source-id: 2a411d78895f3f3aa9ab0e4807c17a13e7f25caf
This commit is contained in:
Patrick Labatut
2021-06-09 15:48:56 -07:00
committed by Facebook GitHub Bot
parent 02650672f6
commit 07da36d4c8
3 changed files with 5 additions and 3 deletions

View File

@@ -142,7 +142,7 @@ def sample_points_from_meshes(
def _rand_barycentric_coords(
size1, size2, dtype, device
size1, size2, dtype: torch.dtype, device: torch.device
) -> Tuple[torch.Tensor, torch.Tensor, torch.Tensor]:
"""
Helper function to generate random barycentric coordinates which are uniformly