cyto_dl import comment out

AllenCell · Jan 10, 2024 · a68c221 · a68c221
1 parent 6346e23
commit a68c221
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/src/allencell_ml_segmenter/_tests/services/test_prediction_service.py b/src/allencell_ml_segmenter/_tests/services/test_prediction_service.py
@@ -14,7 +14,6 @@
 from allencell_ml_segmenter.services.prediction_service import (
     PredictionService,
 )
-import cyto_dl
 
 
 @pytest.fixture

diff --git a/src/allencell_ml_segmenter/services/prediction_service.py b/src/allencell_ml_segmenter/services/prediction_service.py
@@ -8,7 +8,7 @@
 from allencell_ml_segmenter.prediction.model import PredictionModel
 from pathlib import Path
 from typing import List, Any
-from cyto_dl.api.model import CytoDLModel
+#from cyto_dl.api.model import CytoDLModel
 from napari.utils.notifications import show_warning
 
 

diff --git a/src/allencell_ml_segmenter/services/training_service.py b/src/allencell_ml_segmenter/services/training_service.py
@@ -3,7 +3,7 @@
 from allencell_ml_segmenter.core.subscriber import Subscriber
 from allencell_ml_segmenter.core.event import Event
 
-from cyto_dl.api.model import CytoDLModel
+#from cyto_dl.api.model import CytoDLModel
 
 # from lightning.pytorch.callbacks import Callback