Reformat with black and isort

obss · Jun 4, 2024 · 9f26c66 · 9f26c66
1 parent 9c8aff5
commit 9f26c66
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sahi/models/mmdet.py b/sahi/models/mmdet.py
@@ -191,8 +191,8 @@ def has_mask(self):
         Returns if model output contains segmentation mask
         """
         # has_mask = self.model.model.with_mask
-        train_pipeline = self.model.cfg["train_dataloader"]['dataset']['pipeline']
-        has_mask = any(isinstance(item, dict) and any('mask' in key for key in item.keys()) for item in train_pipeline)
+        train_pipeline = self.model.cfg["train_dataloader"]["dataset"]["pipeline"]
+        has_mask = any(isinstance(item, dict) and any("mask" in key for key in item.keys()) for item in train_pipeline)
         return has_mask
 
     @property