From 42d447ddc49e348a5a6008d908e2aab0c70da18a Mon Sep 17 00:00:00 2001 From: Eugene Liu Date: Thu, 16 Jan 2025 13:19:28 +0000 Subject: [PATCH] Skip tests for unsupported models: add check for DFine --- tests/integration/cli/test_cli.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/integration/cli/test_cli.py b/tests/integration/cli/test_cli.py index 649bbf6a4c..0de2a49092 100644 --- a/tests/integration/cli/test_cli.py +++ b/tests/integration/cli/test_cli.py @@ -252,6 +252,9 @@ def test_otx_e2e( if "dino" in model_name: return # DINO is not supported. + if "dfine" in model_name: + return # DFine is not supported. + if "rtdetr" in model_name: return # RT-DETR currently is not supported. @@ -331,6 +334,9 @@ def test_otx_explain_e2e( if "dino" in model_name: pytest.skip("DINO is not supported.") + if "dfine" in model_name: + pytest.skip("DFine is not supported.") + if "maskrcnn_r50_tv" in model_name: pytest.skip("MaskRCNN R50 Torchvision model doesn't support explain.") elif "rtdetr" in recipe: