Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

test: replacing `tmpdir` with `tmp_path` in `tests_pytorch/serve` (#19650)

This commit is contained in:
Felix Hirwa Nshuti 2024-03-17 11:19:34 +05:30 committed by GitHub
parent 754734a654
commit e57a7bff92
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/tests_pytorch/serve/test_servable_module_validator.py
View File

@ -37,10 +37,10 @@ def test_servable_module_validator():
@pytest.mark.flaky(reruns=3) @pytest.mark.flaky(reruns=3)
def test_servable_module_validator_with_trainer(tmpdir): def test_servable_module_validator_with_trainer(tmp_path):
callback = ServableModuleValidator() callback = ServableModuleValidator()
trainer = Trainer( trainer = Trainer(
default_root_dir=tmpdir, default_root_dir=tmp_path,
max_epochs=1, max_epochs=1,
limit_train_batches=2, limit_train_batches=2,
limit_val_batches=0, limit_val_batches=0,