black reformatting

PTG-Kitware · Nov 21, 2024 · 72c0a03 · 72c0a03
1 parent e66e122
commit 72c0a03
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/angel_system/global_step_prediction/global_step_predictor.py b/angel_system/global_step_prediction/global_step_predictor.py
@@ -170,8 +170,8 @@ def compute_average_TP_activations(
 
             for ann_id in coco_truth.index.anns:
                 ann = coco_truth.index.anns[ann_id]
-                if ann['category_id'] == activity_id:
-                    referenced_img_id = ann['image_id']
+                if ann["category_id"] == activity_id:
+                    referenced_img_id = ann["image_id"]
                     pred_ann = coco_train.annots(image_id=referenced_img_id).objs
                     assert len(pred_ann) == 1
                     pred_ann = pred_ann[0]

diff --git a/angel_system/global_step_prediction/run_expirement.py b/angel_system/global_step_prediction/run_expirement.py
@@ -53,7 +53,9 @@ def run_inference_all_vids(
         if avg_probs is not None:
             step_predictor.get_average_TP_activations_from_array(avg_probs)
         else:
-            avg_probs = step_predictor.compute_average_TP_activations(coco_train, coco_test)
+            avg_probs = step_predictor.compute_average_TP_activations(
+                coco_train, coco_test
+            )
             save_file = (
                 code_dir
                 / "model_files"
@@ -111,6 +113,7 @@ def get_unique(activity_ids):
             granular_or_broad="granular",
         )
 
+
 @click.command(context_settings={"help_option_names": ["-h", "--help"]})
 @click.argument(
     "medical_task",