Remove TestCase dependency for test_models_detection_anchor_utils.py (#4207)

8d540228 · Nicolas Hug · GitHub · 693e0ae8 · 8d540228
Unverified Commit 8d540228 authored Jul 26, 2021 by Nicolas Hug Committed by GitHub Jul 26, 2021
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

test/test_models_detection_anchor_utils.py test/test_models_detection_anchor_utils.py +2 -2

No files found.
--- a/test/test_models_detection_anchor_utils.py
+++ b/test/test_models_detection_anchor_utils.py
 import torch
-from common_utils import TestCase, assert_equal
+from common_utils import assert_equal
 from torchvision.models.detection.anchor_utils import AnchorGenerator, DefaultBoxGenerator
 from torchvision.models.detection.image_list import ImageList
 import pytest


-class Tester(TestCase):
+class Tester:
    def test_incorrect_anchors(self):
        incorrect_sizes = ((2, 4, 8), (32, 8), )
        incorrect_aspects = (0.5, 1.0)