Skip to content

Commit eb082b2

Browse files
authored
fix: mlflow tmp_dir in /tmp (#20642)
1 parent bd9d114 commit eb082b2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/lightning/pytorch/loggers/mlflow.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -369,7 +369,7 @@ def _scan_and_log_checkpoints(self, checkpoint_callback: ModelCheckpoint) -> Non
369369
self.experiment.log_artifact(self._run_id, p, artifact_path)
370370

371371
# Create a temporary directory to log on mlflow
372-
with tempfile.TemporaryDirectory(prefix="test", suffix="test", dir=os.getcwd()) as tmp_dir:
372+
with tempfile.TemporaryDirectory() as tmp_dir:
373373
# Log the metadata
374374
with open(f"{tmp_dir}/metadata.yaml", "w") as tmp_file_metadata:
375375
yaml.dump(metadata, tmp_file_metadata, default_flow_style=False)

0 commit comments

Comments
 (0)