* create dir first before writing to files

HYLcool · HYLcool · commit 55480e3a602d · 2025-06-17T17:29:51.000+08:00
diff --git a/trinity/buffer/ray_wrapper.py b/trinity/buffer/ray_wrapper.py
@@ -142,6 +142,8 @@ def __init__(self, storage_config: StorageConfig, config: BufferConfig) -> None:
             raise ValueError(
                 f"File path must end with '.json' or '.jsonl', got {storage_config.path}"
             )
+        path_dir = os.path.dirname(storage_config.path)
+        os.makedirs(path_dir, exist_ok=True)
         self.file = open(storage_config.path, "a", encoding="utf-8")
         self.encoder = _Encoder(ensure_ascii=False)
 

Original file line number	Diff line number	Diff line change
`@@ -142,6 +142,8 @@ def __init__(self, storage_config: StorageConfig, config: BufferConfig) -> None:`
`142`	`142`	`raise ValueError(`
`143`	`143`	`f"File path must end with '.json' or '.jsonl', got {storage_config.path}"`
`144`	`144`	`)`
	`145`	`+ path_dir = os.path.dirname(storage_config.path)`
	`146`	`+ os.makedirs(path_dir, exist_ok=True)`
`145`	`147`	`self.file = open(storage_config.path, "a", encoding="utf-8")`
`146`	`148`	`self.encoder = _Encoder(ensure_ascii=False)`
`147`	`149`