Skip to content

Commit

Permalink
name changing
Browse files Browse the repository at this point in the history
  • Loading branch information
h-mayorquin committed Jun 20, 2024
1 parent be59dbe commit a98e81a
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 7 deletions.
4 changes: 2 additions & 2 deletions src/spikeinterface/preprocessing/preprocessinglist.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@
CenterRecording,
center,
)
from .scale import ScaleTouV, scale_to_uV
from .scale import ScaleTouVRecording, scale_to_uV

from .whiten import WhitenRecording, whiten, compute_whitening_matrix
from .rectify import RectifyRecording, rectify
Expand Down Expand Up @@ -56,7 +56,7 @@
ScaleRecording,
CenterRecording,
ZScoreRecording,
ScaleTouV,
ScaleTouVRecording,
# decorrelation stuff
WhitenRecording,
# re-reference
Expand Down
4 changes: 2 additions & 2 deletions src/spikeinterface/preprocessing/scale.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
from spikeinterface.preprocessing.basepreprocessor import BasePreprocessor


class ScaleTouV(BasePreprocessor):
class ScaleTouVRecording(BasePreprocessor):
"""
Scale raw traces to microvolts (µV).
Expand Down Expand Up @@ -43,4 +43,4 @@ def __init__(self, recording: BaseRecording):
)


scale_to_uV = ScaleTouV
scale_to_uV = ScaleTouVRecording
6 changes: 3 additions & 3 deletions src/spikeinterface/preprocessing/tests/test_scaling.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import pytest
import numpy as np
from spikeinterface.core.testing_tools import generate_recording
from spikeinterface.preprocessing import ScaleTouV # Replace 'your_module' with your actual module name
from spikeinterface.preprocessing import ScaleTouVRecording # Replace 'your_module' with your actual module name


def test_scale_to_uv():
Expand All @@ -22,7 +22,7 @@ def test_scale_to_uv():
recording.set_channel_offsets(offsets)

# Apply the preprocessor
scaled_recording = ScaleTouV(recording=recording)
scaled_recording = ScaleTouVRecording(recording=recording)

# Check if the traces are indeed scaled
expected_traces = recording.get_traces(return_scaled=True)
Expand All @@ -33,4 +33,4 @@ def test_scale_to_uv():
# Test for the error when recording doesn't have scaleable traces
recording.set_channel_gains(None) # Remove gains to make traces unscaleable
with pytest.raises(AssertionError):
ScaleTouV(recording)
ScaleTouVRecording(recording)

0 comments on commit a98e81a

Please sign in to comment.