File tree Expand file tree Collapse file tree 3 files changed +26
-5
lines changed Expand file tree Collapse file tree 3 files changed +26
-5
lines changed Original file line number Diff line number Diff line change @@ -924,7 +924,14 @@ def get_openephys_probe_data(ephys_recording_key):
924
924
session_dir = find_full_path (get_ephys_root_data_dir (),
925
925
get_session_directory (ephys_recording_key ))
926
926
loaded_oe = openephys .OpenEphys (session_dir )
927
- return loaded_oe .probes [inserted_probe_serial_number ]
927
+ probe_data = loaded_oe .probes [inserted_probe_serial_number ]
928
+
929
+ # explicitly garbage collect "loaded_oe"
930
+ # as these may have large memory footprint and may not be cleared fast enough
931
+ del loaded_oe
932
+ gc .collect ()
933
+
934
+ return probe_data
928
935
929
936
930
937
def get_neuropixels_channel2electrode_map (ephys_recording_key , acq_software ):
Original file line number Diff line number Diff line change @@ -867,10 +867,17 @@ def get_spikeglx_meta_filepath(ephys_recording_key):
867
867
def get_openephys_probe_data (ephys_recording_key ):
868
868
inserted_probe_serial_number = (ProbeInsertion * probe .Probe
869
869
& ephys_recording_key ).fetch1 ('probe' )
870
- sess_dir = find_full_path (get_ephys_root_data_dir (),
870
+ session_dir = find_full_path (get_ephys_root_data_dir (),
871
871
get_session_directory (ephys_recording_key ))
872
- loaded_oe = openephys .OpenEphys (sess_dir )
873
- return loaded_oe .probes [inserted_probe_serial_number ]
872
+ loaded_oe = openephys .OpenEphys (session_dir )
873
+ probe_data = loaded_oe .probes [inserted_probe_serial_number ]
874
+
875
+ # explicitly garbage collect "loaded_oe"
876
+ # as these may have large memory footprint and may not be cleared fast enough
877
+ del loaded_oe
878
+ gc .collect ()
879
+
880
+ return probe_data
874
881
875
882
876
883
def get_neuropixels_channel2electrode_map (ephys_recording_key , acq_software ):
Original file line number Diff line number Diff line change @@ -882,7 +882,14 @@ def get_openephys_probe_data(ephys_recording_key):
882
882
session_dir = find_full_path (get_ephys_root_data_dir (),
883
883
get_session_directory (ephys_recording_key ))
884
884
loaded_oe = openephys .OpenEphys (session_dir )
885
- return loaded_oe .probes [inserted_probe_serial_number ]
885
+ probe_data = loaded_oe .probes [inserted_probe_serial_number ]
886
+
887
+ # explicitly garbage collect "loaded_oe"
888
+ # as these may have large memory footprint and may not be cleared fast enough
889
+ del loaded_oe
890
+ gc .collect ()
891
+
892
+ return probe_data
886
893
887
894
888
895
def get_neuropixels_channel2electrode_map (ephys_recording_key , acq_software ):
You can’t perform that action at this time.
0 commit comments