From ad71c6645a8d0d1f9a2660af9df356ac2125f692 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Francesc=20Mart=C3=AD=20Escofet?= Date: Fri, 14 Jun 2024 14:49:41 +0200 Subject: [PATCH] run pchs --- tests/test_learner.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/test_learner.py b/tests/test_learner.py index 2afe3c6..404fbeb 100644 --- a/tests/test_learner.py +++ b/tests/test_learner.py @@ -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():