Skip to content

Commit

Permalink
add: add first new baseline, now that the workflow runs
Browse files Browse the repository at this point in the history
  • Loading branch information
jnsbck committed Nov 21, 2024
1 parent baa98e7 commit c82862d
Showing 1 changed file with 8 additions and 8 deletions.
16 changes: 8 additions & 8 deletions tests/test_regression.py
Original file line number Diff line number Diff line change
Expand Up @@ -110,7 +110,7 @@ def append_to_json(fpath, test_name, input_kwargs, runtimes):


class compare_to_baseline:
def __init__(self, baseline_iters=1, test_iters=1):
def __init__(self, baseline_iters=3, test_iters=1):
self.baseline_iters = baseline_iters
self.test_iters = test_iters

Expand Down Expand Up @@ -186,13 +186,13 @@ def build_net(num_cells, artificial=True, connect=True, connection_prob=0.0):
(
# Test a single SWC cell with both solvers.
pytest.param(1, False, False, 0.0, "jaxley.stone"),
# pytest.param(1, False, False, 0.0, "jax.sparse"),
# # Test a network of SWC cells with both solvers.
# pytest.param(10, False, True, 0.1, "jaxley.stone"),
# pytest.param(10, False, True, 0.1, "jax.sparse"),
# # Test a larger network of smaller neurons with both solvers.
# pytest.param(1000, True, True, 0.001, "jaxley.stone"),
# pytest.param(1000, True, True, 0.001, "jax.sparse"),
pytest.param(1, False, False, 0.0, "jax.sparse"),
# Test a network of SWC cells with both solvers.
pytest.param(10, False, True, 0.1, "jaxley.stone"),
pytest.param(10, False, True, 0.1, "jax.sparse"),
# Test a larger network of smaller neurons with both solvers.
pytest.param(1000, True, True, 0.001, "jaxley.stone"),
pytest.param(1000, True, True, 0.001, "jax.sparse"),
),
)
@compare_to_baseline(baseline_iters=3)
Expand Down

0 comments on commit c82862d

Please sign in to comment.