From a8e4e6b4aeb529917a51bdf187c1fcc518956f54 Mon Sep 17 00:00:00 2001 From: Xu Yang Date: Mon, 14 Oct 2024 17:53:04 +0800 Subject: [PATCH] fix: fix a small bug in cache using module name and function name as unique folder name (#429) --- rdagent/core/utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/rdagent/core/utils.py b/rdagent/core/utils.py index 86aada12..8fb1a7de 100644 --- a/rdagent/core/utils.py +++ b/rdagent/core/utils.py @@ -125,7 +125,9 @@ def cache_with_pickle(hash_func: Callable, post_process_func: Callable | None = @functools.wraps(func) def cache_wrapper(*args: Any, **kwargs: Any) -> Any: if RD_AGENT_SETTINGS.cache_with_pickle: - target_folder = Path(RD_AGENT_SETTINGS.pickle_cache_folder_path_str) / func.__module__ + target_folder = ( + Path(RD_AGENT_SETTINGS.pickle_cache_folder_path_str) / f"{func.__module__}.{func.__name__}" + ) target_folder.mkdir(parents=True, exist_ok=True) hash_key = hash_func(*args, **kwargs) if hash_key is not None and (target_folder / (hash_key + ".pkl")).exists():