fix: tests

RolnickLab · Dec 6, 2024 · afb946f · afb946f
1 parent 2cfd591
commit afb946f
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/trapdata/api/tests/test_api.py b/trapdata/api/tests/test_api.py
@@ -7,12 +7,11 @@
 from trapdata.api.api import (
     PIPELINE_CHOICES,
     PipelineChoice,
-    PipelineConfig,
     PipelineRequest,
     PipelineResponse,
-    SourceImageRequest,
     app,
 )
+from trapdata.api.schemas import PipelineConfig, SourceImageRequest
 from trapdata.api.tests.image_server import StaticFileTestServer
 from trapdata.ml.models.classification import SpeciesClassifier
 from trapdata.tests import TEST_IMAGES_BASE_PATH