Skip to content

Commit fc8d40f

Browse files
author
Hannah DeFazio
committed
black reformatting
1 parent e6aba07 commit fc8d40f

File tree

2 files changed

+64
-10
lines changed

2 files changed

+64
-10
lines changed

angel_system/data/medical/data_paths.py

+62-4
Original file line numberDiff line numberDiff line change
@@ -98,8 +98,36 @@
9898
"test": [
9999
f"{m2_lab_bbn_data_dir}/tq_{x}"
100100
for x in [
101-
1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
102-
16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30
101+
1,
102+
2,
103+
3,
104+
4,
105+
5,
106+
6,
107+
7,
108+
8,
109+
9,
110+
10,
111+
11,
112+
12,
113+
13,
114+
14,
115+
15,
116+
16,
117+
17,
118+
18,
119+
19,
120+
20,
121+
21,
122+
22,
123+
23,
124+
24,
125+
25,
126+
26,
127+
27,
128+
28,
129+
29,
130+
30,
103131
]
104132
],
105133
}
@@ -114,8 +142,38 @@
114142
"test": [
115143
f"{m2_kitware_data_dir}/kitware_m2_video_{x}_extracted"
116144
for x in [
117-
1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17,
118-
18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32
145+
1,
146+
2,
147+
3,
148+
4,
149+
5,
150+
6,
151+
7,
152+
8,
153+
9,
154+
10,
155+
11,
156+
12,
157+
13,
158+
14,
159+
15,
160+
16,
161+
17,
162+
18,
163+
19,
164+
20,
165+
21,
166+
22,
167+
23,
168+
24,
169+
25,
170+
26,
171+
27,
172+
28,
173+
29,
174+
30,
175+
31,
176+
32,
119177
]
120178
],
121179
}

angel_system/data/medical/load_bbn_data.py

+2-6
Original file line numberDiff line numberDiff line change
@@ -258,14 +258,10 @@ def bbn_activity_txt_to_csv(root_dir, output_dir):
258258
"""
259259
print(f"{root_dir}/*/*_action_labels_by_frame.txt")
260260

261-
action_fns = glob.glob(
262-
f"{root_dir}/*/*.action_labels_by_frame.txt"
263-
)
261+
action_fns = glob.glob(f"{root_dir}/*/*.action_labels_by_frame.txt")
264262
if not action_fns:
265263
# Lab videos
266-
action_fns = glob.glob(
267-
f"{root_dir}/*/*_skills_frame.txt"
268-
)
264+
action_fns = glob.glob(f"{root_dir}/*/*_skills_frame.txt")
269265
if not action_fns:
270266
warnings.warn(f"No text annotations found in {root_dir}")
271267
return

0 commit comments

Comments
 (0)