black reformatting

PTG-Kitware · Feb 15, 2024 · fc8d40f · fc8d40f
1 parent e6aba07
commit fc8d40f
Show file tree

Hide file tree

Showing 2 changed files with 64 additions and 10 deletions.
diff --git a/angel_system/data/medical/data_paths.py b/angel_system/data/medical/data_paths.py
@@ -98,8 +98,36 @@
     "test": [
         f"{m2_lab_bbn_data_dir}/tq_{x}"
         for x in [
-            1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
-            16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30
+            1,
+            2,
+            3,
+            4,
+            5,
+            6,
+            7,
+            8,
+            9,
+            10,
+            11,
+            12,
+            13,
+            14,
+            15,
+            16,
+            17,
+            18,
+            19,
+            20,
+            21,
+            22,
+            23,
+            24,
+            25,
+            26,
+            27,
+            28,
+            29,
+            30,
         ]
     ],
 }
@@ -114,8 +142,38 @@
     "test": [
         f"{m2_kitware_data_dir}/kitware_m2_video_{x}_extracted"
         for x in [
-            1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17,
-            18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32
+            1,
+            2,
+            3,
+            4,
+            5,
+            6,
+            7,
+            8,
+            9,
+            10,
+            11,
+            12,
+            13,
+            14,
+            15,
+            16,
+            17,
+            18,
+            19,
+            20,
+            21,
+            22,
+            23,
+            24,
+            25,
+            26,
+            27,
+            28,
+            29,
+            30,
+            31,
+            32,
         ]
     ],
 }

diff --git a/angel_system/data/medical/load_bbn_data.py b/angel_system/data/medical/load_bbn_data.py
@@ -258,14 +258,10 @@ def bbn_activity_txt_to_csv(root_dir, output_dir):
     """
     print(f"{root_dir}/*/*_action_labels_by_frame.txt")
 
-    action_fns = glob.glob(
-        f"{root_dir}/*/*.action_labels_by_frame.txt"
-    )
+    action_fns = glob.glob(f"{root_dir}/*/*.action_labels_by_frame.txt")
     if not action_fns:
         # Lab videos
-        action_fns = glob.glob(
-            f"{root_dir}/*/*_skills_frame.txt"
-        )
+        action_fns = glob.glob(f"{root_dir}/*/*_skills_frame.txt")
     if not action_fns:
         warnings.warn(f"No text annotations found in {root_dir}")
         return