Skip to content

Commit 59dc274

Browse files
committed
Clean up print statements in cnmfe
1 parent ddb5281 commit 59dc274

File tree

1 file changed

+1
-11
lines changed

1 file changed

+1
-11
lines changed

mesmerize_core/algorithms/cnmfe.py

Lines changed: 1 addition & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44
from caiman.source_extraction.cnmf import cnmf as cnmf
55
from caiman.source_extraction.cnmf.params import CNMFParams
66
from caiman.paths import decode_mmap_filename_dict
7-
from caiman.base.movies import get_file_size
87
import traceback
98
from pathlib import Path, PurePosixPath
109
from shutil import move as move_file
@@ -17,12 +16,11 @@
1716
ensure_server,
1817
save_projections_parallel,
1918
save_c_order_mmap_parallel,
20-
estimate_n_pixels_per_process
2119
)
2220
else: # when running with local backend
2321
from ..batch_utils import set_parent_raw_data_path, load_batch
2422
from ..utils import IS_WINDOWS
25-
from ._utils import ensure_server, save_projections_parallel, save_c_order_mmap_parallel, estimate_n_pixels_per_process
23+
from ._utils import ensure_server, save_projections_parallel, save_c_order_mmap_parallel
2624

2725

2826
def run_algo(batch_path, uuid, data_path: str = None, dview=None):
@@ -65,14 +63,6 @@ def run_algo(batch_path, uuid, data_path: str = None, dview=None):
6563

6664
if save_new_mmap:
6765
print("making memmap")
68-
dims, T = get_file_size(input_movie_path, var_name_hdf5=cnmfe_params_dict.data['var_name_hdf5'])
69-
assert isinstance(T, int)
70-
print('Movie dims:', dims)
71-
print('N frames:', T)
72-
print('N processes:', n_processes)
73-
print('N pixels per process:', chunk_size := estimate_n_pixels_per_process(n_processes, T, dims))
74-
print('Columns per chunk:', max(chunk_size // dims[0], 1))
75-
breakpoint()
7666
fname_new = save_c_order_mmap_parallel(
7767
input_movie_path,
7868
base_name=f"{uuid}_cnmf-memmap_",

0 commit comments

Comments
 (0)