Skip to content

Commit d9dd6df

Browse files
committed
fixup! Allow bool, str (case insensitive), BaseProfiler
1 parent 7bc42ad commit d9dd6df

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/trainer/test_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1198,5 +1198,5 @@ def test_trainer_profiler_correct_args(profiler, expected):
11981198

11991199
def test_trainer_profiler_incorrect_str_arg():
12001200
with pytest.raises(ValueError, match=r"When passing string value for the `profiler` parameter of"
1201-
"` Trainer`, it can only be 'simple' or 'advanced'"):
1201+
" `Trainer`, it can only be 'simple' or 'advanced'"):
12021202
Trainer(profiler="unknown_profiler")

0 commit comments

Comments
 (0)