Skip to content

Commit

Permalink
local import
Browse files Browse the repository at this point in the history
  • Loading branch information
alejoe91 committed Sep 15, 2024
1 parent dcedbb3 commit 9000ce1
Showing 1 changed file with 2 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,6 @@
import pytest
import shutil
import numpy as np
import pandas as pd

from spikeinterface.core import generate_ground_truth_recording
from spikeinterface.core import create_sorting_analyzer, load_sorting_analyzer
Expand Down Expand Up @@ -117,6 +116,8 @@ def _check_one(self, sorting_analyzer, extension_class, params):
with the passed parameters, and check the output is not empty, the extension
exists and `select_units()` method works.
"""
import pandas as pd

if extension_class.need_job_kwargs:
job_kwargs = dict(n_jobs=2, chunk_duration="1s", progress_bar=True)
else:
Expand Down

0 comments on commit 9000ce1

Please sign in to comment.