Skip to content

Commit d69d578

Browse files
committed
Use num_samples instead of durations in hybrid tools
1 parent e3d0311 commit d69d578

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/spikeinterface/generation/hybrid_tools.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -400,6 +400,7 @@ def generate_hybrid_recording(
400400
num_segments = recording.get_num_segments()
401401
dtype = recording.dtype
402402
durations = np.array([recording.get_duration(segment_index) for segment_index in range(num_segments)])
403+
num_samples = np.array([recording.get_num_samples(segment_index) for segment_index in range(num_segments)])
403404
channel_locations = probe.contact_positions
404405

405406
assert (
@@ -548,7 +549,7 @@ def generate_hybrid_recording(
548549
displacement_vectors=displacement_vectors,
549550
displacement_sampling_frequency=displacement_sampling_frequency,
550551
displacement_unit_factor=displacement_unit_factor,
551-
num_samples=(np.array(durations) * sampling_frequency).astype("int64"),
552+
num_samples=num_samples.astype("int64"),
552553
amplitude_factor=amplitude_factor,
553554
)
554555

0 commit comments

Comments
 (0)