Skip to content

Commit

Permalink
Update src/pynwb/testing/mock/ecephys.py
Browse files Browse the repository at this point in the history
Co-authored-by: Ben Dichter <[email protected]>
  • Loading branch information
h-mayorquin and bendichter authored Mar 31, 2024
1 parent 6959e0f commit 3de731d
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions src/pynwb/testing/mock/ecephys.py
Original file line number Diff line number Diff line change
Expand Up @@ -125,15 +125,13 @@ def mock_SpikeEventSeries(
def mock_Units(
num_units: int = 10,
max_spikes_per_unit: int = 10,
seed: Optional[int] = None,
seed: int = 0,
nwbfile: Optional[NWBFile] = None,
) -> Units:

units_table = Units()
units_table.add_column(name="unit_name", description="a readable identifier for the unit")

if seed is None:
seed = 0
rng = np.random.default_rng(seed=seed)

times = rng.random(size=(num_units, max_spikes_per_unit)).cumsum(axis=1)
Expand Down

0 comments on commit 3de731d

Please sign in to comment.