diff --git a/python/single_cell_model.cpp b/python/single_cell_model.cpp index b71987cf3e256d6d944fe1ec342bcccc1effff1b..dfba5158ff45faa0aa829cb66582f27cc0a976cc 100644 --- a/python/single_cell_model.cpp +++ b/python/single_cell_model.cpp @@ -171,7 +171,7 @@ public: gprop_.add_ion(ion, valence, int_con, ext_con, rev_pot); } - void run(double tfinal) { + void run(double tfinal, double dt) { single_cell_recipe rec(cell_, probes_, gprop_); auto domdec = arb::partition_load_balance(rec, ctx_); @@ -201,7 +201,7 @@ public: } }); - sim_->run(tfinal, 0.025); + sim_->run(tfinal, dt); run_ = true; } @@ -233,7 +233,11 @@ void register_single_cell(pybind11::module& m) { model .def(pybind11::init<arb::cable_cell>(), "cell"_a, "Initialise a single cell model for a cable cell.") - .def("run", &single_cell_model::run, "tfinal"_a, "Run model from t=0 to t=tfinal ms.") + .def("run", + &single_cell_model::run, + "tfinal"_a, + "dt"_a = 0.025, + "Run model from t=0 to t=tfinal ms.") .def("probe", [](single_cell_model& m, const char* what, const char* where, double frequency) { m.probe(what, where, frequency);},