Skip to content

Commit

Permalink
Fix tests
Browse files Browse the repository at this point in the history
  • Loading branch information
alejoe91 committed Aug 29, 2024
1 parent 77e7367 commit 7bfc818
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 2 deletions.
5 changes: 3 additions & 2 deletions src/spikeinterface_pipelines/spikesorting/params.py
Original file line number Diff line number Diff line change
Expand Up @@ -129,9 +129,10 @@ class Kilosort4Model(BaseModel):
save_extra_kwargs: bool = Field(default=False, description="If True, additional kwargs are saved to the output")
skip_kilosort_preprocessing: bool = Field(default=False, description="Can optionally skip the internal kilosort preprocessing")
scaleproc: Union[None, int] = Field(default=None, description="int16 scaling of whitened data, if None set to 200.")
save_preprocessed_copy: bool = Field(default=False, description="save a pre-processed copy of the data (including drift correction) to temp_wh.dat in the results directory and format Phy output to use that copy of the data")
torch_device: str = Field(default="auto", description="Select the torch device auto/cuda/cpu")
# bad_channels: Optional[List[int]] = Field(default=None, description="List of bad channels to exclude from spike detection and clustering.")
# THESE 2 PARAMS REUQIRE A NEW RELEASE OD SPIKEINTERFACE
# ist[int]] = Field(default=None, description="List of bad channels to exclude from spike detection and clustering.")
# save_preprocessed_copy: bool = Field(default=False, description="save a pre-processed copy of the data (including drift correction) to temp_wh.dat in the results directory and format Phy output to use that copy of the data")



Expand Down
1 change: 1 addition & 0 deletions tests/test_pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,7 @@ def _generate_gt_recording():
analyzer.compute(
[
"random_spikes",
"waveforms",
"templates",
"noise_levels",
"spike_amplitudes",
Expand Down

0 comments on commit 7bfc818

Please sign in to comment.