Accommodate 'test' stim files that aren't actual tests

AllenInstitute · Oct 14, 2023 · 8f59fde · 8f59fde
1 parent 91ffece
commit 8f59fde
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/npc_lims/paths/s3.py b/src/npc_lims/paths/s3.py
@@ -328,7 +328,10 @@ def get_hdf5_stim_files_from_s3(
         )
     file_glob = f"*_{session.subject}_{session.date.replace('-', '')}_??????.hdf5"
     files = [StimFile(path, session) for path in root.glob(file_glob)]
-
+
+    test_glob = file_glob.replace(str(session.subject), "test")
+    files += [StimFile(path, session) for path in root.glob(test_glob) if str(session.subject) in path.as_posix()]
+
     # no empty files:
     files = [f for f in files if f.size > 0]