Skip to content

Commit 4f8134a

Browse files
authored
fix: fix a small bug in cache using module name and function name as unique folder name (#429)
1 parent 5715668 commit 4f8134a

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

rdagent/core/utils.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -125,7 +125,9 @@ def cache_decorator(func: Callable) -> Callable:
125125
@functools.wraps(func)
126126
def cache_wrapper(*args: Any, **kwargs: Any) -> Any:
127127
if RD_AGENT_SETTINGS.cache_with_pickle:
128-
target_folder = Path(RD_AGENT_SETTINGS.pickle_cache_folder_path_str) / func.__module__
128+
target_folder = (
129+
Path(RD_AGENT_SETTINGS.pickle_cache_folder_path_str) / f"{func.__module__}.{func.__name__}"
130+
)
129131
target_folder.mkdir(parents=True, exist_ok=True)
130132
hash_key = hash_func(*args, **kwargs)
131133
if hash_key is not None and (target_folder / (hash_key + ".pkl")).exists():

0 commit comments

Comments
 (0)