From df670d8fb79bc4379a3f2eb2a36de2c7c8dc36f8 Mon Sep 17 00:00:00 2001 From: Eugene Liu Date: Thu, 16 Jan 2025 11:34:49 +0000 Subject: [PATCH] Skip tests for unsupported models: add check for D-Fine --- tests/integration/api/test_xai.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/integration/api/test_xai.py b/tests/integration/api/test_xai.py index 8a565233f4..3da3476886 100644 --- a/tests/integration/api/test_xai.py +++ b/tests/integration/api/test_xai.py @@ -51,6 +51,10 @@ def test_forward_explain( # TODO(Eugene): maskdino not support yet. pytest.skip(f"There's issue with inst-seg: {model_name}. Skip for now.") + if "dfine" in model_name: + # TODO(Eugene): dfine not support yet. + pytest.skip(f"There's issue with dfine: {model_name}. Skip for now.") + if "dino" in model_name: pytest.skip("DINO is not supported.")