From 5815dd403bf856b78dba7cd0b11ff752fb674034 Mon Sep 17 00:00:00 2001 From: Cemberk Date: Thu, 20 Feb 2025 14:46:18 +0000 Subject: [PATCH] import missing --- tests/models/grounding_dino/test_modeling_grounding_dino.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/models/grounding_dino/test_modeling_grounding_dino.py b/tests/models/grounding_dino/test_modeling_grounding_dino.py index 2e8e33ee7abe..8a5874e0ea41 100644 --- a/tests/models/grounding_dino/test_modeling_grounding_dino.py +++ b/tests/models/grounding_dino/test_modeling_grounding_dino.py @@ -34,6 +34,7 @@ require_vision, slow, torch_device, + skipIfRocm, ) from ...test_configuration_common import ConfigTester