Skip to content

Commit 97adde9

Browse files
kaushikb11lexierule
authored andcommitted
Fix tensorboard test
1 parent 8500c53 commit 97adde9

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

tests/loggers/test_tensorboard.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,12 +25,10 @@
2525
from pytorch_lightning import Trainer
2626
from pytorch_lightning.loggers import TensorBoardLogger
2727
from tests.helpers import BoringModel
28+
from tests.helpers.runif import RunIf
2829

2930

30-
@pytest.mark.skipif(
31-
LooseVersion(torch.__version__) < LooseVersion("1.5.0"),
32-
reason="Minimal PT version is set to 1.5",
33-
)
31+
@RunIf(min_torch="1.5.0")
3432
def test_tensorboard_hparams_reload(tmpdir):
3533

3634
class CustomModel(BoringModel):

0 commit comments

Comments
 (0)