From 969f056e1d565c7d974a3164fe40eef8a02501e0 Mon Sep 17 00:00:00 2001 From: Ben Dichter Date: Sun, 12 May 2024 20:41:32 -0400 Subject: [PATCH] Update testing.py types (#324) --- src/roiextractors/testing.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/roiextractors/testing.py b/src/roiextractors/testing.py index f052e8f8..56ee9753 100644 --- a/src/roiextractors/testing.py +++ b/src/roiextractors/testing.py @@ -1,7 +1,7 @@ """Testing utilities for the roiextractors package.""" from collections.abc import Iterable -from typing import Tuple, Optional +from typing import Tuple, Optional, List import numpy as np from numpy.testing import assert_array_equal, assert_array_almost_equal @@ -52,9 +52,9 @@ def generate_dummy_imaging_extractor( num_rows: int = 10, num_columns: int = 10, num_channels: int = 1, - sampling_frequency: float = 30, + sampling_frequency: float = 30.0, dtype: DtypeType = "uint16", - channel_names: Optional[list] = None, + channel_names: Optional[List[str]] = None, ): """Generate a dummy imaging extractor for testing.