cache kwargs for function

ecmwf-projects · Sep 2, 2024 · b98c1f2 · b98c1f2
1 parent 683d4b0
commit b98c1f2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cacholote/cache.py b/cacholote/cache.py
@@ -45,7 +45,7 @@ def _decode_and_update(
     return result
 
 
-def cacheable(func: F) -> F:
+def cacheable(func: F, **cache_kwargs) -> F:
     """Make a function cacheable."""
 
     @functools.wraps(func)
@@ -56,7 +56,7 @@ def wrapper(*args: Any, **kwargs: Any) -> Any:
             return func(*args, **kwargs)
 
         try:
-            hexdigest = encode._hexdigestify_python_call(func, *args, **kwargs)
+            hexdigest = encode._hexdigestify_python_call(func, *args, **kwargs, **cache_kwargs)
         except encode.EncodeError as ex:
             if settings.return_cache_entry:
                 raise ex