Merge branch 'main' of github.com:cwindolf/spike-psvae into main

cwindolf · Nov 7, 2023 · 842f079 · 842f079
2 parents 2ef7a3a + 65e8eea
commit 842f079
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/dartsort/transform/all_transformers.py b/src/dartsort/transform/all_transformers.py
@@ -1,6 +1,6 @@
 from .amplitudes import AmplitudeVector, MaxAmplitude
 from .enforce_decrease import EnforceDecrease
-from .localize import Localization
+from .localize import Localization, PointSourceLocalization
 from .single_channel_denoiser import SingleChannelWaveformDenoiser
 from .temporal_pca import TemporalPCADenoiser, TemporalPCAFeaturizer
 from .transform_base import Waveform
@@ -14,6 +14,7 @@
     TemporalPCADenoiser,
     TemporalPCAFeaturizer,
     Localization,
+    PointSourceLocalization,
 ]
 
 transformers_by_class_name = {cls.__name__: cls for cls in all_transformers}
diff --git a/src/dartsort/transform/localize.py b/src/dartsort/transform/localize.py
@@ -66,3 +66,5 @@ def transform(self, waveforms, max_channels=None):
             ]
         )
         return localizations
+
+PointSourceLocalization = Localization
-Original file line number
+Diff line change
@@ Expand Up / @@ -66,3 +66,5 @@ def transform(self, waveforms, max_channels=None): @@
                 ]
             )
             return localizations
+    PointSourceLocalization = Localization