Fix for flake8, black

labelme/ai/__init__.py

@@ -1,9 +1,11 @@
 import collections
 
-from .models.segment_anything import SegmentAnythingModel
+from .models.segment_anything import SegmentAnythingModel  # NOQA
 
 
-Model = collections.namedtuple("Model", ["name", "encoder_weight", "decoder_weight"])
+Model = collections.namedtuple(
+    "Model", ["name", "encoder_weight", "decoder_weight"]
+)
 
 Weight = collections.namedtuple("Weight", ["url", "md5"])
 

labelme/app.py

@@ -753,7 +753,9 @@ class MainWindow(QtWidgets.QMainWindow):
         selectAiModel.setDefaultWidget(QtWidgets.QWidget())
         selectAiModel.defaultWidget().setLayout(QtWidgets.QVBoxLayout())
         self._selectAiModelComboBox = QtWidgets.QComboBox()
-        selectAiModel.defaultWidget().layout().addWidget(self._selectAiModelComboBox)
+        selectAiModel.defaultWidget().layout().addWidget(
+            self._selectAiModelComboBox
+        )
         self._selectAiModelComboBox.addItems([model.name for model in MODELS])
         self._selectAiModelComboBox.setCurrentIndex(1)
         self._selectAiModelComboBox.setEnabled(False)

labelme/widgets/canvas.py

@@ -140,11 +140,11 @@ class Canvas(QtWidgets.QWidget):
             logger.debug("Initializing AI model: %r" % model.name)
             self._ai_model = labelme.ai.SegmentAnythingModel(
                 name=model.name,
-                encoder_path = gdown.cached_download(
+                encoder_path=gdown.cached_download(
                     url=model.encoder_weight.url,
                     md5=model.encoder_weight.md5,
                 ),
-                decoder_path = gdown.cached_download(
+                decoder_path=gdown.cached_download(
                     url=model.decoder_weight.url,
                     md5=model.decoder_weight.md5,
                 ),