Update test_event_csrmm.py

brainpy · Jun 9, 2024 · bbb57e4 · bbb57e4
1 parent 4a16207
commit bbb57e4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/brainpy/_src/math/event/tests/test_event_csrmm.py b/brainpy/_src/math/event/tests/test_event_csrmm.py
@@ -4,6 +4,7 @@
 from functools import partial
 
 import jax
+import pytest
 from absl.testing import parameterized
 
 import brainpy as bp