From d9dd6dff569d70d023e69a3cbb84733cd7b01671 Mon Sep 17 00:00:00 2001 From: Dusan Drevicky Date: Fri, 2 Oct 2020 10:24:10 +0200 Subject: [PATCH] fixup! Allow bool, str (case insensitive), BaseProfiler --- tests/trainer/test_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/trainer/test_trainer.py b/tests/trainer/test_trainer.py index 3d9f6a75690aae..666ac92c0d7a25 100644 --- a/tests/trainer/test_trainer.py +++ b/tests/trainer/test_trainer.py @@ -1198,5 +1198,5 @@ def test_trainer_profiler_correct_args(profiler, expected): def test_trainer_profiler_incorrect_str_arg(): with pytest.raises(ValueError, match=r"When passing string value for the `profiler` parameter of" - "` Trainer`, it can only be 'simple' or 'advanced'"): + " `Trainer`, it can only be 'simple' or 'advanced'"): Trainer(profiler="unknown_profiler")