From d098beb7a7f92ee226de97b1b7905ee735aeed56 Mon Sep 17 00:00:00 2001 From: Jeremy Reizenstein Date: Thu, 4 Sep 2025 08:31:32 -0700 Subject: [PATCH] allow python 3.12 Summary: Remove use of distutils Reviewed By: MichaelRamamonjisoa Differential Revision: D81594552 fbshipit-source-id: 4e979d5e03ea873bd09bc2b674b7e6480b9c6d65 --- tests/test_rotation_conversions.py | 2 -- tests/test_so3.py | 2 -- 2 files changed, 4 deletions(-) diff --git a/tests/test_rotation_conversions.py b/tests/test_rotation_conversions.py index 87bd250f..1ed99453 100644 --- a/tests/test_rotation_conversions.py +++ b/tests/test_rotation_conversions.py @@ -8,7 +8,6 @@ import itertools import math import unittest -from distutils.version import LooseVersion from typing import Optional, Union 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 ) - @unittest.skipIf(LooseVersion(torch.__version__) < "1.9", "recent torchscript only") def test_scriptable(self): torch.jit.script(axis_angle_to_matrix) torch.jit.script(axis_angle_to_quaternion) diff --git a/tests/test_so3.py b/tests/test_so3.py index 5819957c..617cd313 100644 --- a/tests/test_so3.py +++ b/tests/test_so3.py @@ -7,7 +7,6 @@ import math import unittest -from distutils.version import LooseVersion import numpy as np import torch @@ -255,7 +254,6 @@ class TestSO3(TestCaseMixin, unittest.TestCase): # all grad values have to be finite self.assertTrue(torch.isfinite(r.grad).all()) - @unittest.skipIf(LooseVersion(torch.__version__) < "1.9", "recent torchscript only") def test_scriptable(self): torch.jit.script(so3_exp_map) torch.jit.script(so3_log_map)