Skip to content

Commit c60932b

Browse files
committed
skipif no kineto
1 parent a68f904 commit c60932b

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/profiler/test_profiler.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -526,6 +526,7 @@ def test_trainer_profiler_incorrect_str_arg():
526526
Trainer(profiler="unknown_profiler")
527527

528528

529+
@pytest.mark.skipif(not _KINETO_AVAILABLE, reason="Requires PyTorch Profiler Kineto")
529530
@pytest.mark.parametrize(
530531
["trainer_config", "trainer_fn"],
531532
[

0 commit comments

Comments
 (0)