Skip to content

Commit 038e765

Browse files
committed
fix printing
1 parent 4db64a4 commit 038e765

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

bnd/pipeline/nwbtools/multiprobe_kilosort_interface.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -288,10 +288,12 @@ def __init__(
288288
"""
289289
self.session_path = ksorted_folder_path.parent.parent
290290
self.recording_to_process = ksorted_folder_path.name[-2:] # g0 or g1
291-
outputs_paths = list(Path(ksorted_folder_path).glob("*/spike_times.npy"))
291+
outputs_paths = list(Path(ksorted_folder_path).glob("**/spike_times.npy"))
292292
self.sorter_output_paths = [path.parent for path in outputs_paths]
293293
self.custom_map = custom_map
294294

295+
# breakpoint()
296+
295297
if not len(self.sorter_output_paths):
296298
raise ValueError("Selected recording does not have kilosort output")
297299

bnd/pipeline/pyaldata.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -802,7 +802,6 @@ def _partition_and_save_to_mat(self):
802802
# Check if size exceeds matlab 5 format
803803
nbytes = _get_nbytes_from_recarray(recarray=data_array)
804804
num_partitions = int(np.ceil(nbytes / (2**31)))
805-
logger.info(f"Session ({nbytes / 2**30:.2f} GB) exceeds matlab 5 format (2 GB) ")
806805
recarray_size = len(data_array)
807806

808807
assert num_partitions >= 1 and num_partitions < len(data_array) # General checks
@@ -818,6 +817,9 @@ def _partition_and_save_to_mat(self):
818817
return
819818
else:
820819
# Partition array
820+
logger.info(
821+
f"Session ({nbytes / 2**30:.2f} GB) exceeds matlab 5 format (2 GB) "
822+
)
821823

822824
logger.info(f"Partitioning array into {num_partitions} chunks...")
823825
partition_sizes = [

0 commit comments

Comments
 (0)