From 9e97cf51139f7df5edb903984be1fde9b3652e06 Mon Sep 17 00:00:00 2001 From: haosenwang1018 Date: Tue, 24 Feb 2026 21:58:00 +0000 Subject: [PATCH] fix: use context manager for file read in wandb_util open().read() without a context manager leaks the file descriptor until garbage collection. Use 'with open()' to ensure prompt cleanup. --- turbodiffusion/imaginaire/utils/wandb_util.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/turbodiffusion/imaginaire/utils/wandb_util.py b/turbodiffusion/imaginaire/utils/wandb_util.py index 7914e44..f1ee41e 100644 --- a/turbodiffusion/imaginaire/utils/wandb_util.py +++ b/turbodiffusion/imaginaire/utils/wandb_util.py @@ -89,7 +89,8 @@ def _read_wandb_id(config_job: JobConfig, config_checkpoint: CheckpointConfig) - wandb_id = None wandb_id_path = f"{config_job.path_local}/wandb_id.txt" if os.path.isfile(wandb_id_path): - wandb_id = open(wandb_id_path).read().strip() + with open(wandb_id_path) as f: + wandb_id = f.read().strip() return wandb_id