diff --git a/tests/agent/test_auxiliary_relay.py b/tests/agent/test_auxiliary_relay.py index 8866167b42ae..0902f0fb91c3 100644 --- a/tests/agent/test_auxiliary_relay.py +++ b/tests/agent/test_auxiliary_relay.py @@ -2,6 +2,8 @@ from types import SimpleNamespace import pytest +pytest.importorskip("nemo_relay") + from agent import auxiliary_client, relay_llm, relay_runtime diff --git a/tests/agent/test_relay_llm.py b/tests/agent/test_relay_llm.py index 4cd9484d2bdb..24a63781568b 100644 --- a/tests/agent/test_relay_llm.py +++ b/tests/agent/test_relay_llm.py @@ -8,6 +8,8 @@ from types import SimpleNamespace import pytest +pytest.importorskip("nemo_relay") + from agent import relay_llm, relay_runtime diff --git a/tests/agent/test_relay_tools.py b/tests/agent/test_relay_tools.py index 36ca4322ff0e..8aed80eacdee 100644 --- a/tests/agent/test_relay_tools.py +++ b/tests/agent/test_relay_tools.py @@ -4,6 +4,8 @@ from __future__ import annotations import pytest +pytest.importorskip("nemo_relay") + from agent import relay_runtime, relay_tools