allow python 3.12

Summary: Remove use of distutils

Reviewed By: MichaelRamamonjisoa

Differential Revision: D81594552

fbshipit-source-id: 4e979d5e03ea873bd09bc2b674b7e6480b9c6d65
This commit is contained in:
Jeremy Reizenstein 2025-09-04 08:31:32 -07:00 committed by Facebook GitHub Bot
parent dd068703d1
commit d098beb7a7
2 changed files with 0 additions and 4 deletions

View File

@ -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)

View File

@ -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)