Skip to content

Commit cc21f06

Browse files
authored
Apply suggestions from code review
1 parent 5769eff commit cc21f06

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

src/spikeinterface/core/sortinganalyzer.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1349,12 +1349,13 @@ def compute_several_extensions(self, extensions, save=True, verbose=False, **job
13491349
)
13501350
t_end = perf_counter()
13511351
# for pipeline node extensions we can only track the runtime of the run_node_pipeline
1352-
runtime_s = np.round(t_end - t_start, 1)
1352+
runtime_s = t_end - t_start
13531353

13541354
for r, result in enumerate(results):
13551355
extension_name, variable_name = result_routage[r]
13561356
extension_instances[extension_name].data[variable_name] = result
13571357
extension_instances[extension_name].run_info["runtime_s"] = runtime_s
1358+
extension_instances[extension_name].run_info["run_completed"] = True
13581359

13591360
for extension_name, extension_instance in extension_instances.items():
13601361
self.extensions[extension_name] = extension_instance
@@ -2008,7 +2009,7 @@ def run(self, save=True, **kwargs):
20082009
t_start = perf_counter()
20092010
self._run(**kwargs)
20102011
t_end = perf_counter()
2011-
self.run_info["runtime_s"] = np.round(t_end - t_start, 1)
2012+
self.run_info["runtime_s"] = t_end - t_start
20122013

20132014
if save and not self.sorting_analyzer.is_read_only():
20142015
self._save_data(**kwargs)
@@ -2020,7 +2021,6 @@ def save(self, **kwargs):
20202021
self._save_params()
20212022
self._save_importing_provenance()
20222023
self._save_data(**kwargs)
2023-
self.run_info["run_completed"] = True
20242024
self._save_run_info()
20252025

20262026
def _save_data(self, **kwargs):

0 commit comments

Comments
 (0)