Update imports in test to access internal function.

SpikeInterface · Mar 28, 2024 · 4ce4791 · 4ce4791
1 parent 9eab1a7
commit 4ce4791
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/test_io/test_spikegadgets.py b/tests/test_io/test_spikegadgets.py
@@ -3,10 +3,8 @@
 
 import pytest
 
-from probeinterface import (
-    read_spikegadgets,
-    parse_spikegadgets_header,
-)
+from probeinterface import read_spikegadgets
+from probeinterface.io import parse_spikegadgets_header
 
 data_path = Path(__file__).absolute().parent.parent / "data" / "spikegadgets"
 test_file = "SpikeGadgets_test_data_2xNpix1.0_20240318_173658_header_only.rec"