From 452f23b6168ef951ef35e00d20fdaff43eedeec9 Mon Sep 17 00:00:00 2001 From: Alessio Buccino Date: Thu, 6 Nov 2025 17:58:18 +0100 Subject: [PATCH] tiny doc fixes --- doc/api.rst | 11 +++++++---- src/spikeinterface/generation/drift_tools.py | 2 ++ 2 files changed, 9 insertions(+), 4 deletions(-) diff --git a/doc/api.rst b/doc/api.rst index b7fef13502..2aa09767a9 100755 --- a/doc/api.rst +++ b/doc/api.rst @@ -397,7 +397,7 @@ spikeinterface.generation Core ~~~~ -.. currentmodule:: spikeinterface.generation +.. automodule:: spikeinterface.generation .. autofunction:: generate_recording .. autofunction:: generate_sorting @@ -418,7 +418,8 @@ Core Drift ~~~~~ -.. currentmodule:: spikeinterface.generation +.. automodule:: spikeinterface.generation + :no-index: .. autofunction:: generate_drifting_recording .. autofunction:: generate_displacement_vector @@ -432,7 +433,8 @@ Drift Hybrid ~~~~~~ -.. currentmodule:: spikeinterface.generation +.. automodule:: spikeinterface.generation + :no-index: .. autofunction:: generate_hybrid_recording .. autofunction:: estimate_templates_from_recording @@ -448,7 +450,8 @@ Hybrid Noise ~~~~~ -.. currentmodule:: spikeinterface.generation +.. automodule:: spikeinterface.generation + :no-index: .. autofunction:: generate_noise diff --git a/src/spikeinterface/generation/drift_tools.py b/src/spikeinterface/generation/drift_tools.py index 70c9803921..0989afe126 100644 --- a/src/spikeinterface/generation/drift_tools.py +++ b/src/spikeinterface/generation/drift_tools.py @@ -7,6 +7,8 @@ from numpy.typing import ArrayLike from spikeinterface.core import BaseRecording, BaseRecordingSegment, BaseSorting, Templates +from probeinterface import Probe + def interpolate_templates(templates_array, source_locations, dest_locations, interpolation_method="cubic"): """