Skip to content

Commit

Permalink
lint + comment out cyto_dl for ci
Browse files Browse the repository at this point in the history
  • Loading branch information
yrkim98 committed Feb 22, 2024
1 parent 0f0029d commit ae4a173
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 3 deletions.
1 change: 1 addition & 0 deletions src/allencell_ml_segmenter/prediction/view.py
Original file line number Diff line number Diff line change
Expand Up @@ -99,6 +99,7 @@ def doWork(self):
# test service
self._prediction_model.dispatch_prediction()
# TODO Need way to set result images to show after prediction complete and refresh viewer.

def getTypeOfWork(self):
return "Prediction"

Expand Down
6 changes: 4 additions & 2 deletions src/allencell_ml_segmenter/services/prediction_service.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
from pathlib import Path
from typing import Union, Dict, List

from cyto_dl.api.model import CytoDLModel
# from cyto_dl.api.model import CytoDLModel
from napari.utils.notifications import show_warning


Expand Down Expand Up @@ -49,7 +49,9 @@ def _predict_model(self, _: Event) -> None:
):
cyto_api: CytoDLModel = CytoDLModel()
cyto_api.load_config_from_file(
self._experiments_model.get_train_config_path(self._experiments_model.get_experiment_name())
self._experiments_model.get_train_config_path(
self._experiments_model.get_experiment_name()
)
)
# We must override the config to set up predictions correctly
cyto_api.override_config(
Expand Down
2 changes: 1 addition & 1 deletion src/allencell_ml_segmenter/services/training_service.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down

0 comments on commit ae4a173

Please sign in to comment.