Skip to content

Commit

Permalink
add dsec to conftest
Browse files Browse the repository at this point in the history
  • Loading branch information
ntolley committed Oct 25, 2023
1 parent 05ab75d commit 95ff6ce
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions hnn_core/tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -75,8 +75,8 @@ def pytest_runtest_setup(item):
def run_hnn_core_fixture():
def _run_hnn_core_fixture(backend=None, n_procs=None, n_jobs=1,
reduced=False, record_vsec=False,
record_isec=False, postproc=False,
electrode_array=None):
record_isec=False, record_dcell=False,
postproc=False, electrode_array=None):
hnn_core_root = op.dirname(hnn_core.__file__)

# default params
Expand Down Expand Up @@ -105,15 +105,18 @@ def _run_hnn_core_fixture(backend=None, n_procs=None, n_jobs=1,
with MPIBackend(n_procs=n_procs, mpi_cmd='mpiexec'):
dpls = simulate_dipole(net, record_vsec=record_vsec,
record_isec=record_isec,
record_dcell=record_dcell,
postproc=postproc, tstop=tstop)
elif backend == 'joblib':
with JoblibBackend(n_jobs=n_jobs):
dpls = simulate_dipole(net, record_vsec=record_vsec,
record_isec=record_isec,
record_dcell=record_dcell,
postproc=postproc, tstop=tstop)
else:
dpls = simulate_dipole(net, record_vsec=record_vsec,
record_isec=record_isec,
record_dcell=record_dcell,
postproc=postproc, tstop=tstop)

# check that the network object is picklable after the simulation
Expand Down

0 comments on commit 95ff6ce

Please sign in to comment.