diff --git a/src/spikeinterface/generation/drifting_generator.py b/src/spikeinterface/generation/drifting_generator.py index e4ab723ae3..39fb2b4a81 100644 --- a/src/spikeinterface/generation/drifting_generator.py +++ b/src/spikeinterface/generation/drifting_generator.py @@ -298,7 +298,7 @@ def calculate_displacement_unit_factor( factors = 1 - factors f = np.abs(non_rigid_gradient) - print("f", f) + displacement_unit_factor = factors * (1 - f) + f return displacement_unit_factor