remove print statements

Reviewed By: gkioxari

Differential Revision: D19834684

fbshipit-source-id: 553dbf84d1062149b4915d313fc0f96eb047798c
This commit is contained in:
Jeremy Reizenstein
2020-02-11 15:14:42 -08:00
committed by Facebook Github Bot
parent 09992a388f
commit 23bb27956a
2 changed files with 5 additions and 3 deletions

View File

@@ -2,6 +2,7 @@
# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
import math
import warnings
import torch
@@ -663,7 +664,7 @@ def _check_valid_rotation_matrix(R, tol: float = 1e-7):
Returns:
None
Prints an warning if R is an invalid rotation matrix. Else return.
Emits a warning if R is an invalid rotation matrix.
"""
N = R.shape[0]
eye = torch.eye(3, dtype=R.dtype, device=R.device)
@@ -673,5 +674,5 @@ def _check_valid_rotation_matrix(R, tol: float = 1e-7):
no_distortion = torch.allclose(det_R, torch.ones_like(det_R))
if not (orthogonal and no_distortion):
msg = "R is not a valid rotation matrix"
print(msg)
warnings.warn(msg)
return