[Fix]: fix polarmix UT (#2310)

5d88bef6 · Sun Jiahao · GitHub · 60d848b3 · 5d88bef6
Unverified Commit 5d88bef6 authored Feb 28, 2023 by Sun Jiahao Committed by GitHub Feb 28, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/test_datasets/test_transforms/test_transforms_3d.py tests/test_datasets/test_transforms/test_transforms_3d.py +1 -1

No files found.
--- a/tests/test_datasets/test_transforms/test_transforms_3d.py
+++ b/tests/test_datasets/test_transforms/test_transforms_3d.py
@@ -216,7 +216,7 @@ class TestPolarMix(unittest.TestCase):
            transform = PolarMix(instance_classes=[1.0, 2.0])

        transform = PolarMix(
-            instance_classes=[1, 2],
+            instance_classes=[15, 16, 17],
            swap_ratio=1.0,
            pre_transform=self.pre_transform)
        results = transform.transform(copy.deepcopy(self.results))