Fix bbox unit test (#1914)

open-mmlab · Apr 26, 2022 · de0c103 · de0c103
1 parent 0e59036
commit de0c103
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_ops/test_bbox.py b/tests/test_ops/test_bbox.py
@@ -8,7 +8,7 @@
 
 class TestBBox(object):
 
-    def _test_bbox_overlaps(self, dtype=torch.float, device='cpu'):
+    def _test_bbox_overlaps(self, device='cpu', dtype=torch.float):
         from mmcv.ops import bbox_overlaps
         b1 = torch.tensor([[1.0, 1.0, 3.0, 4.0], [2.0, 2.0, 3.0, 4.0],
                            [7.0, 7.0, 8.0, 8.0]]).to(device).type(dtype)