in1d to isin

ryanharvey1 · Dec 6, 2024 · 94fa6f2 · 94fa6f2
1 parent 5c5c7ee
commit 94fa6f2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_event_triggered_average_fast.py b/tests/test_event_triggered_average_fast.py
@@ -13,7 +13,7 @@ def test_event_triggered_average_fast():
     # 5 random timestamps from 0 to 10
     ts_rand = np.random.choice(ts, 5)
     # set the 5 signals to 5 at the random timestamps
-    x[:, np.in1d(ts, ts_rand)] = 5
+    x[:, np.isin(ts, ts_rand)] = 5
     # make psth of the 5 signals at the random timestamps
     psth = event_triggered_average_fast(
         x,