persistent-function-cache 0.2.1__py3-none-any.whl → 0.2.2__py3-none-any.whl

persistent_cache/main/cache_slot.py

@@ -3,6 +3,7 @@ import pickle
 from collections.abc import Callable, Iterable, Iterator
 from dataclasses import dataclass
 from functools import cached_property
+from inspect import BoundArguments
 from typing import Any
 
 from persistent_cache.models import Path
@@ -18,6 +19,7 @@ class CacheSlot:
     kwargs: dict[str, Any]
     directory: Path
     key_arguments: Iterable[str] | str | None
+    argument_reducers: dict[str, Callable[[Any], Any]] | None
     extra_keys: Any
     key_reducer: type[Reducer] | None
     deep_learning: bool
@@ -60,17 +62,25 @@ class CacheSlot:
 
     @property
     def argument_values(self) -> Iterator[Any]:
-        if self.key_arguments is None:
+        if self.key_arguments is None and self.argument_reducers is None:
             yield from self.args
             yield from self.kwargs.values()
         else:
             arguments = inspect.signature(self.function).bind(*self.args, **self.kwargs)
             arguments.apply_defaults()
+            yield from self.extract_argument_values(arguments)
+
+    def extract_argument_values(self, arguments: BoundArguments) -> Iterator[Any]:
+        if self.key_arguments is not None:
             if isinstance(self.key_arguments, str):
                 yield arguments.arguments.get(self.key_arguments)
             else:
                 for name in self.key_arguments:
                     yield arguments.arguments.get(name)
+        if self.argument_reducers is not None:
+            for argument_name, reducer in self.argument_reducers.items():
+                argument = arguments.arguments.get(argument_name)
+                yield reducer(argument)
 
     @property
     def reducer(self) -> type[Reducer]:

persistent_cache/main/decorator.py

@@ -16,6 +16,7 @@ def cache(
     *,
     cache_directory: Path = Path.cache,
     cache_key_arguments: Iterable[str] | str | None = None,
+    argument_reducers: dict[str, Callable[[Any], Any]] | None = None,
     extra_cache_keys: Iterable[Any] | None = None,
     key_reducer: type[Reducer] = Reducer,
     deep_learning: bool = False,
@@ -29,6 +30,7 @@ def cache(
     *,
     cache_directory: Path = Path.cache,
     cache_key_arguments: Iterable[str] | str | None = None,
+    argument_reducers: dict[str, Callable[[Any], Any]] | None = None,
     extra_cache_keys: Any = None,
     key_reducer: type[Reducer] = Reducer,
     deep_learning: bool = False,
@@ -41,6 +43,7 @@ def cache(  # noqa: PLR0913
     *,
     cache_directory: Path = Path.cache,
     cache_key_arguments: Iterable[str] | str | None = None,
+    argument_reducers: dict[str, Callable[[Any], Any]] | None = None,
     extra_cache_keys: Any = None,
     key_reducer: type[Reducer] | None = None,
     deep_learning: bool = False,
@@ -68,6 +71,7 @@ def cache(  # noqa: PLR0913
                 kwargs,
                 cache_directory,
                 cache_key_arguments,
+                argument_reducers,
                 extra_cache_keys,
                 key_reducer,
                 deep_learning,

{persistent_function_cache-0.2.1.dist-info → persistent_function_cache-0.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: persistent-function-cache
-Version: 0.2.1
+Version: 0.2.2
 Summary: Persistent cache for expensive functions
 Author-email: Quinten Roets <qdr2104@columbia.edu>
 License: MIT

{persistent_function_cache-0.2.1.dist-info → persistent_function_cache-0.2.2.dist-info}/RECORD

@@ -6,8 +6,8 @@ persistent_cache/caches/speedup_deep_learning.py,sha256=bywNXeEZsg8Rs2A1_w-ZrJi9
 persistent_cache/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 persistent_cache/cli/clear_cache.py,sha256=YbKfqzUB9IyEpU-2n0h5BrK9kZKjbayqiMTzjPbDl3s,1541
 persistent_cache/main/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-persistent_cache/main/cache_slot.py,sha256=9zBrcTGJ-oEUryftiGqBks_mdrkX2s_c0b2Pisc5vvU,2716
-persistent_cache/main/decorator.py,sha256=tvrgv_lTxldgxfWnvzJKvDC00oE8eenPY5VdPZxiywM,2439
+persistent_cache/main/cache_slot.py,sha256=MmWBcz798QA8N6CPXXNxYmlGb3HwbG21NbbFcavH6eU,3265
+persistent_cache/main/decorator.py,sha256=lE_0q-KEavCe8IaK8hwuMUxH-tkN1_v-Jgy_dM_8Uxw,2684
 persistent_cache/main/hashing.py,sha256=6GYzUJcqjfjHPWMqPOCEY9FUSH2k2zUGrUza2fKaMTQ,2688
 persistent_cache/models/__init__.py,sha256=1OeLZ6FZ5gAjlerAjc69Vx0AqWLSx6OSqNCpuzkCZQg,23
 persistent_cache/models/path.py,sha256=-HRJ4yqqmBPlax4H0KS8guhndBXNwCnl7WZXhtaWmGQ,441
@@ -15,9 +15,9 @@ persistent_cache/reducers/__init__.py,sha256=hCozFuvrAb_gXLcqMMq92v4KdwBbQugwGwC
 persistent_cache/reducers/base.py,sha256=HsodcuE8Id5oV93_b_RN18I69UWsJ21yazXCwT1f6YU,1428
 persistent_cache/reducers/deep_learning.py,sha256=cyVolwmwm_ZN2gRxPtfJFqiNQlNuq_CrehO7ZSVWeVw,800
 persistent_cache/reducers/speedup_deep_learning.py,sha256=kMkr1467zUsQUPOi_N8tWKWAtKUcFgsk3rh94QhETg0,1970
-persistent_function_cache-0.2.1.dist-info/LICENSE,sha256=ENpNaBSvIV7ifD7iNz_-lBhImbiYowqPzBc5V5R8IhM,1070
-persistent_function_cache-0.2.1.dist-info/METADATA,sha256=E_GxkYLEh9vSXGpVYv9cKAMDTjqHRJDZc6d44WcqQXs,2164
-persistent_function_cache-0.2.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-persistent_function_cache-0.2.1.dist-info/entry_points.txt,sha256=fSZfuL6Wn42mZldojgGyg7dzSPCRVAuZox3w6dLd2ZI,88
-persistent_function_cache-0.2.1.dist-info/top_level.txt,sha256=Jj-Z9wxHRlRyDUxrfC9GeaV5-8hAf8R4LclQFZesxi8,17
-persistent_function_cache-0.2.1.dist-info/RECORD,,
+persistent_function_cache-0.2.2.dist-info/LICENSE,sha256=ENpNaBSvIV7ifD7iNz_-lBhImbiYowqPzBc5V5R8IhM,1070
+persistent_function_cache-0.2.2.dist-info/METADATA,sha256=PoI9EMIFqiZ21RHpHKNctkbdhu7h0LdHWD7TS-8_Krc,2164
+persistent_function_cache-0.2.2.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+persistent_function_cache-0.2.2.dist-info/entry_points.txt,sha256=fSZfuL6Wn42mZldojgGyg7dzSPCRVAuZox3w6dLd2ZI,88
+persistent_function_cache-0.2.2.dist-info/top_level.txt,sha256=Jj-Z9wxHRlRyDUxrfC9GeaV5-8hAf8R4LclQFZesxi8,17
+persistent_function_cache-0.2.2.dist-info/RECORD,,