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)