mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-09-16 01:22:48 +08:00
allow python 3.12
Summary: Remove use of distutils Reviewed By: MichaelRamamonjisoa Differential Revision: D81594552 fbshipit-source-id: 4e979d5e03ea873bd09bc2b674b7e6480b9c6d65
This commit is contained in:
parent
dd068703d1
commit
d098beb7a7
@ -8,7 +8,6 @@
|
|||||||
import itertools
|
import itertools
|
||||||
import math
|
import math
|
||||||
import unittest
|
import unittest
|
||||||
from distutils.version import LooseVersion
|
|
||||||
from typing import Optional, Union
|
from typing import Optional, Union
|
||||||
|
|
||||||
import numpy as np
|
import numpy as np
|
||||||
@ -271,7 +270,6 @@ class TestRotationConversion(TestCaseMixin, unittest.TestCase):
|
|||||||
torch.matmul(r, r.permute(0, 2, 1)), torch.eye(3).expand_as(r), atol=1e-6
|
torch.matmul(r, r.permute(0, 2, 1)), torch.eye(3).expand_as(r), atol=1e-6
|
||||||
)
|
)
|
||||||
|
|
||||||
@unittest.skipIf(LooseVersion(torch.__version__) < "1.9", "recent torchscript only")
|
|
||||||
def test_scriptable(self):
|
def test_scriptable(self):
|
||||||
torch.jit.script(axis_angle_to_matrix)
|
torch.jit.script(axis_angle_to_matrix)
|
||||||
torch.jit.script(axis_angle_to_quaternion)
|
torch.jit.script(axis_angle_to_quaternion)
|
||||||
|
@ -7,7 +7,6 @@
|
|||||||
|
|
||||||
import math
|
import math
|
||||||
import unittest
|
import unittest
|
||||||
from distutils.version import LooseVersion
|
|
||||||
|
|
||||||
import numpy as np
|
import numpy as np
|
||||||
import torch
|
import torch
|
||||||
@ -255,7 +254,6 @@ class TestSO3(TestCaseMixin, unittest.TestCase):
|
|||||||
# all grad values have to be finite
|
# all grad values have to be finite
|
||||||
self.assertTrue(torch.isfinite(r.grad).all())
|
self.assertTrue(torch.isfinite(r.grad).all())
|
||||||
|
|
||||||
@unittest.skipIf(LooseVersion(torch.__version__) < "1.9", "recent torchscript only")
|
|
||||||
def test_scriptable(self):
|
def test_scriptable(self):
|
||||||
torch.jit.script(so3_exp_map)
|
torch.jit.script(so3_exp_map)
|
||||||
torch.jit.script(so3_log_map)
|
torch.jit.script(so3_log_map)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user