Skip to content

Commit 3b187cb

Browse files
committed
update: update params in test/tutorials
1 parent 297e261 commit 3b187cb

File tree

1 file changed

+28
-16
lines changed

1 file changed

+28
-16
lines changed

tests/conftest.py

Lines changed: 28 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -105,35 +105,47 @@ def populate_lfp(pipeline, insert_upstreams):
105105
@pytest.fixture(scope="session")
106106
def insert_clustering_task(pipeline, populate_ephys_recording):
107107
ephys = pipeline["ephys"]
108-
params_ks = {
109-
"fs": 30000,
110-
"fshigh": 150,
108+
params = {}
109+
params["SI_PREPROCESSING_METHOD"] = "CatGT"
110+
params["SI_SORTING_PARAMS"] = {
111111
"minfr_goodchannels": 0.1,
112-
"Th": [10, 4],
113112
"lam": 10,
114113
"AUCsplit": 0.9,
115114
"minFR": 0.02,
116115
"momentum": [20, 400],
117116
"sigmaMask": 30,
118-
"ThPr": 8,
119-
"spkTh": -6,
120-
"reorder": 1,
121-
"nskip": 25,
122-
"GPU": 1,
123-
"Nfilt": 1024,
124117
"nfilt_factor": 4,
125118
"ntbuff": 64,
126-
"whiteningRange": 32,
127-
"nSkipCov": 25,
128119
"scaleproc": 200,
129120
"nPCs": 3,
130-
"useRAM": 0,
131121
}
122+
params["SI_POSTPROCESSING_PARAMS"] = {
123+
"extensions": {
124+
"random_spikes": {},
125+
"waveforms": {},
126+
"templates": {},
127+
"noise_levels": {},
128+
# "amplitude_scalings": {},
129+
"correlograms": {},
130+
"isi_histograms": {},
131+
"principal_components": {"n_components": 5, "mode": "by_channel_local"},
132+
"spike_amplitudes": {},
133+
"spike_locations": {},
134+
"template_metrics": {"include_multi_channel_metrics": True},
135+
"template_similarity": {},
136+
"unit_locations": {},
137+
"quality_metrics": {},
138+
},
139+
"job_kwargs": {"n_jobs": 0.8, "chunk_duration": "1s"},
140+
"export_to_phy": True,
141+
"export_report": True,
142+
}
143+
132144
ephys.ClusteringParamSet.insert_new_params(
133-
clustering_method="kilosort2",
145+
clustering_method="kilosort2.5",
146+
paramset_desc="Default parameter set for Kilosort2.0 with SpikeInterface",
147+
params=params,
134148
paramset_idx=0,
135-
params=params_ks,
136-
paramset_desc="Spike sorting using Kilosort2",
137149
)
138150

139151
session_key = dict(subject="subject5", session_datetime="2023-01-01 00:00:00")

0 commit comments

Comments
 (0)