Skip to content

Commit

Permalink
run pchs
Browse files Browse the repository at this point in the history
  • Loading branch information
FrancescMartiEscofetQC committed Jun 14, 2024
1 parent 1a93bfa commit ad71c66
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion tests/test_learner.py
Original file line number Diff line number Diff line change
Expand Up @@ -315,7 +315,9 @@ def test_learner_twins(metalearner, reference_value, twins_data, rng):
@pytest.mark.parametrize("n_variants", [2, 5])
@pytest.mark.parametrize("is_classification", [True, False])
@pytest.mark.parametrize("is_oos", [True, False])
def test_learner_evaluate(metalearner, is_classification, rng, n_classes, n_variants, is_oos):
def test_learner_evaluate(
metalearner, is_classification, rng, n_classes, n_variants, is_oos
):
sample_size = 1000
factory = metalearner_factory(metalearner)
if n_variants > 2 and not factory._supports_multi_treatment():
Expand Down

0 comments on commit ad71c66

Please sign in to comment.