diff --git a/tests/test_tasks_ml.py b/tests/test_tasks_ml.py index 57afdab..eb59e47 100644 --- a/tests/test_tasks_ml.py +++ b/tests/test_tasks_ml.py @@ -63,7 +63,6 @@ async def test_apply_allowlist_applies_above_threshold(db): from backend.app.models.tag import image_tag from backend.app.services.tag_service import TagService from backend.app.tasks import ml as ml_tasks - from tests._prediction_helpers import seed_predictions tag = await TagService(db).find_or_create("autohero", TagKind.character) @@ -101,7 +100,6 @@ async def test_apply_allowlist_skips_below_threshold(db): from backend.app.models.tag import image_tag from backend.app.services.tag_service import TagService from backend.app.tasks import ml as ml_tasks - from tests._prediction_helpers import seed_predictions tag = await TagService(db).find_or_create("lowconf", TagKind.character)