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

View File

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