black

Author: C-Achard

napari_cellseg3d/_tests/test_helper.py

@@ -13,4 +13,4 @@ def test_helper(make_napari_viewer):
 
     widget.btnc.click()
 
-    assert len(viewer.window._dock_widgets) == children-1
+    assert len(viewer.window._dock_widgets) == children - 1

napari_cellseg3d/_tests/test_interface.py

@@ -1,5 +1,6 @@
 from napari_cellseg3d.interface import Log
 
+
 def test_log(qtbot):
     log = Log()
     log.print_and_log("test")
@@ -10,4 +11,4 @@ def test_log(qtbot):
 
     assert log.toPlainText() == "\ntest2"
 
-    qtbot.add_widget(log)
+    qtbot.add_widget(log)

napari_cellseg3d/_tests/test_plugin_inference.py

@@ -6,6 +6,7 @@
 from napari_cellseg3d.code_plugins.plugin_model_inference import Inferer
 from napari_cellseg3d.code_models.models.model_test import TestModel
 
+
 def test_inference(make_napari_viewer, qtbot):
 
     im_path = str(Path(__file__).resolve().parent / "res/test.tif")

napari_cellseg3d/_tests/test_training.py

@@ -34,7 +34,7 @@ def test_training(make_napari_viewer, qtbot):
     #################
     MODEL_LIST["test"] = TestModel()
     widget.model_choice.addItem("test")
-    widget.model_choice.setCurrentIndex(len(MODEL_LIST.keys())-1)
+    widget.model_choice.setCurrentIndex(len(MODEL_LIST.keys()) - 1)
 
     # widget.start()
     # assert widget.worker is not None

napari_cellseg3d/code_models/model_workers.py

@@ -455,7 +455,9 @@ def model_output(
         inputs = inputs.to("cpu")
 
         model_output = lambda inputs: post_process_transforms(
-            self.config.model_info.get_model().get_output(model, inputs)  # TODO(cyril) refactor those functions
+            self.config.model_info.get_model().get_output(
+                model, inputs
+            )  # TODO(cyril) refactor those functions
         )
 
         def model_output(inputs):

napari_cellseg3d/code_models/models/model_test.py

@@ -7,7 +7,6 @@ def get_weights_file():
 
 
 class TestModel(nn.Module):
-
     def __init__(self):
         super().__init__()
         self.linear = nn.Linear(1, 1)
@@ -24,6 +23,7 @@ def get_output(self, _, input):
     def get_validation(self, val_inputs):
         return val_inputs
 
+
 # if __name__ == "__main__":
 #
 #     model = TestModel()
@@ -33,4 +33,4 @@ def get_validation(self, val_inputs):
 #     torch.save(
 #         model.state_dict(),
 #         WEIGHTS_DIR + f"/{get_weights_file()}"
-#     )
+#     )

napari_cellseg3d/code_plugins/plugin_model_training.py

@@ -791,7 +791,6 @@ def start(self):
                 self.data = None
                 raise err
 
-
             model_config = config.ModelInfo(
                 name=self.model_choice.currentText()
             )

napari_cellseg3d/interface.py

@@ -10,6 +10,7 @@
 from qtpy import QtCore
 from qtpy.QtCore import QObject
 from qtpy.QtCore import Qt
+
 # from qtpy.QtCore import QtWarningMsg
 from qtpy.QtCore import QUrl
 from qtpy.QtGui import QCursor