add nelems to check fun

NeurodataWithoutBorders · Aug 8, 2024 · 28acc06 · 28acc06
1 parent ffb8217
commit 28acc06
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/nwbinspector/checks/time_series.py b/src/nwbinspector/checks/time_series.py
@@ -94,9 +94,9 @@ def check_timestamps_ascending(time_series: TimeSeries, nelems=200):
 
 
 @register_check(importance=Importance.BEST_PRACTICE_VIOLATION, neurodata_type=TimeSeries)
-def check_timestamps_with_nans(time_series: TimeSeries):
+def check_timestamps_with_nans(time_series: TimeSeries, nelems=200):
     """Check if there are NaN values in the timestamps array."""
-    if time_series.timestamps is not None and np.isnan(time_series.timestamps).any():
+    if time_series.timestamps is not None and np.isnan(time_series.timestamps[:nelems]).any():
         return InspectorMessage(message=f"{time_series.name} timestamps contain NaN values.")