diff --git a/tests/test_ml_tagger.py b/tests/test_ml_tagger.py index de288d7..e60237d 100644 --- a/tests/test_ml_tagger.py +++ b/tests/test_ml_tagger.py @@ -10,8 +10,8 @@ import pytest from backend.app.services.ml.tagger import ( STORE_FLOOR, SURFACED_CATEGORIES, - TagPrediction, Tagger, + TagPrediction, get_tagger, )