423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-08-02 11:52:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add missing dtype parameter type annotations

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pytorch3d/datasets/r2n2/utils.py
View File

@ -137,7 +137,9 @@ def compute_extrinsic_matrix(azimuth, elevation, distance):
return RT
def read_binvox_coords(f, integer_division=True, dtype=torch.float32):
def read_binvox_coords(
f, integer_division: bool = True, dtype: torch.dtype = torch.float32
):
"""
Copied from meshrcnn codebase:
https://github.com/facebookresearch/meshrcnn/blob/master/shapenet/utils/binvox_torch.py#L5

2
pytorch3d/io/ply_io.py
View File

@ -462,7 +462,7 @@ def _read_ply_element_ascii(f, definition: _PlyElementType):
return data
def _read_raw_array(f, aim: str, length: int, dtype=np.uint8, dtype_size=1):
def _read_raw_array(f, aim: str, length: int, dtype: type = np.uint8, dtype_size=1):
"""
Read [length] elements from a file.

2
pytorch3d/ops/sample_points_from_meshes.py
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