Merge pull request #43 from janelia-cellmap/actions/black

Format Python code with psf/black push
janelia-cellmap · Feb 9, 2024 · 7ba3662 · 7ba3662
2 parents ce71fb5 + 353b8cb
commit 7ba3662
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/dacapo/experiments/trainers/gunpowder_trainer.py b/dacapo/experiments/trainers/gunpowder_trainer.py
@@ -43,7 +43,9 @@ def __init__(self, trainer_config):
         self.clip_raw = trainer_config.clip_raw
 
         # Testing out if calculating multiple times and multiplying is necessary
-        self.add_predictor_nodes_to_dataset = trainer_config.add_predictor_nodes_to_dataset
+        self.add_predictor_nodes_to_dataset = (
+            trainer_config.add_predictor_nodes_to_dataset
+        )
 
         self.scheduler = None
 
@@ -166,7 +168,9 @@ def build_batch_provider(self, datasets, model, task, snapshot_container=None):
             task.predictor,
             gt_key=gt_key,
             target_key=target_key,
-            weights_key=datasets_weight_key if self.add_predictor_nodes_to_dataset else weight_key,
+            weights_key=datasets_weight_key
+            if self.add_predictor_nodes_to_dataset
+            else weight_key,
             mask_key=mask_key,
         )
 

diff --git a/dacapo/experiments/trainers/gunpowder_trainer_config.py b/dacapo/experiments/trainers/gunpowder_trainer_config.py
@@ -32,5 +32,7 @@ class GunpowderTrainerConfig(TrainerConfig):
 
     add_predictor_nodes_to_dataset: Optional[bool] = attr.ib(
         default=True,
-        metadata={"help_text": "Whether to add a predictor node to dataset_source and apply product of weights"}
+        metadata={
+            "help_text": "Whether to add a predictor node to dataset_source and apply product of weights"
+        },
     )