diff --git a/tests/learning/test_learning_api.py b/tests/learning/test_learning_api.py index 93d058f3..6724af9b 100644 --- a/tests/learning/test_learning_api.py +++ b/tests/learning/test_learning_api.py @@ -2,10 +2,15 @@ from __future__ import annotations -from fastapi import FastAPI -from starlette.testclient import TestClient +import pytest -from openjarvis.server.api_routes import learning_router +fastapi = pytest.importorskip("fastapi") +starlette = pytest.importorskip("starlette") + +from fastapi import FastAPI # noqa: E402 +from starlette.testclient import TestClient # noqa: E402 + +from openjarvis.server.api_routes import learning_router # noqa: E402 def _make_app() -> FastAPI: diff --git a/tests/test_rust_bridge.py b/tests/test_rust_bridge.py index f48152e9..74102ae2 100644 --- a/tests/test_rust_bridge.py +++ b/tests/test_rust_bridge.py @@ -3,9 +3,6 @@ from __future__ import annotations import json -import os - -import pytest class TestGetRustModule: