Skip to content

Commit d47be56

Browse files
committed
refactor: 🎨 refactor PostProcessing
1 parent 7eaefa4 commit d47be56

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

element_array_ephys/spike_sorting/si_spike_sorting.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -253,15 +253,14 @@ class PostProcessing(dj.Imported):
253253
def make(self, key):
254254
execution_time = datetime.utcnow()
255255

256-
# Load recording object.
256+
# Load recording & sorting object.
257257
clustering_method, output_dir, params = (
258258
ephys.ClusteringTask * ephys.ClusteringParamSet & key
259259
).fetch1("clustering_method", "clustering_output_dir", "params")
260260
output_dir = find_full_path(ephys.get_ephys_root_data_dir(), output_dir)
261-
262-
# Get sorter method and create output directory.
263261
sorter_name = clustering_method.replace(".", "_")
264262
output_dir = find_full_path(ephys.get_ephys_root_data_dir(), output_dir)
263+
265264
recording_file = output_dir / sorter_name / "recording" / "si_recording.pkl"
266265
sorting_file = output_dir / sorter_name / "spike_sorting" / "si_sorting.pkl"
267266

@@ -301,14 +300,13 @@ def make(self, key):
301300
_ = si.postprocessing.compute_principal_components(
302301
waveform_extractor=we, **params.get("SI_QUALITY_METRICS_PARAMS", None)
303302
)
303+
metrics = si.qualitymetrics.compute_quality_metrics(waveform_extractor=we)
304+
304305
# Save the output (metrics.csv to the output dir)
305306
metrics_output_dir = output_dir / sorter_name / "metrics"
306307
metrics_output_dir.mkdir(parents=True, exist_ok=True)
307-
308-
metrics = si.qualitymetrics.compute_quality_metrics(waveform_extractor=we)
309308
metrics.to_csv(metrics_output_dir / "metrics.csv")
310309

311-
# Save results
312310
self.insert1(
313311
{
314312
**key,

0 commit comments

Comments
 (0)