Skip to content

GitLab

Unverified Commit 5d88bef6 authored by Sun Jiahao's avatar Sun Jiahao Committed by GitHub
Browse files

[Fix]: fix polarmix UT (#2310)

parent 60d848b3
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_datasets/test_transforms/test_transforms_3d.py
View file @ 5d88bef6
...@@ -216,7 +216,7 @@ class TestPolarMix(unittest.TestCase): ...@@ -216,7 +216,7 @@ class TestPolarMix(unittest.TestCase):
transform = PolarMix(instance_classes=[1.0, 2.0]) transform = PolarMix(instance_classes=[1.0, 2.0])
transform = PolarMix( transform = PolarMix(
instance_classes=[1, 2], instance_classes=[15, 16, 17],
swap_ratio=1.0, swap_ratio=1.0,
pre_transform=self.pre_transform) pre_transform=self.pre_transform)
results = transform.transform(copy.deepcopy(self.results)) results = transform.transform(copy.deepcopy(self.results))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment