checkpointer 2.14.8__tar.gz → 2.14.9__tar.gz

{checkpointer-2.14.8 → checkpointer-2.14.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: checkpointer
-Version: 2.14.8
+Version: 2.14.9
 Summary: checkpointer adds code-aware caching to Python functions, maintaining correctness and speeding up execution as your code changes.
 Project-URL: Repository, https://github.com/Reddan/checkpointer.git
 Author: Hampus Hallman

{checkpointer-2.14.8 → checkpointer-2.14.9}/checkpointer/checkpoint.py

@@ -150,6 +150,8 @@ class CachedFunction(Generic[Fn]):
 
   @property
   def fn(self) -> Fn:
+    if self.bound:
+      return self.ident.fn.__get__(self.bound[0], self.bound[0].__class__)  # type: ignore
     return self.ident.fn  # type: ignore
 
   @cached_property
@@ -201,17 +203,16 @@ class CachedFunction(Generic[Fn]):
   def is_expired(self, call_hash: str) -> bool:
     return not self.storage.exists(call_hash) or self.storage.expired(call_hash)
 
-  def _call(self: CachedFunction[Callable[P, R]], args: tuple, kw: dict, rerun=False, cached=False) -> R:
-    full_args = self.bound + args
+  def _call(self: CachedFunction[Callable[P, R]], args: tuple, kw: dict, cached=True, rerun=False) -> R:
     params = self.ident.checkpointer
-    if not params.when and not cached:
-      return self.fn(*full_args, **kw)
+    if not cached:
+      return self.fn(*args, **kw)
     call_hash = self._get_call_hash(args, kw)
     call_id = f"{self.storage.fn_id()}/{call_hash}"
 
     if rerun or self.is_expired(call_hash):
       print_checkpoint(params.verbosity >= 1, "MEMORIZING", call_id, "blue")
-      data = self.fn(*full_args, **kw)
+      data = self.fn(*args, **kw)
       if iscoroutine(data):
         return self._store_coroutine(call_hash, data)
       return self.storage.store(call_hash, data)
@@ -225,16 +226,19 @@ class CachedFunction(Generic[Fn]):
     except (EOFError, FileNotFoundError):
       pass
     print_checkpoint(params.verbosity >= 1, "CORRUPTED", call_id, "yellow")
-    return self._call(args, kw, True, cached)
+    return self._call(args, kw, rerun=True)
 
   def __call__(self: CachedFunction[Callable[P, R]], *args: P.args, **kw: P.kwargs) -> R:
-    return self._call(args, kw)
+    return self._call(args, kw, self.ident.checkpointer.when)
 
   def cached(self: CachedFunction[Callable[P, R]], *args: P.args, **kw: P.kwargs) -> R:
-    return self._call(args, kw, False, True)
+    return self._call(args, kw, True)
+
+  def uncached(self: CachedFunction[Callable[P, R]], *args: P.args, **kw: P.kwargs) -> R:
+    return self._call(args, kw, False)
 
   def rerun(self: CachedFunction[Callable[P, R]], *args: P.args, **kw: P.kwargs) -> R:
-    return self._call(args, kw, True, True)
+    return self._call(args, kw, rerun=True)
 
   def exists(self: CachedFunction[Callable[P, R]], *args: P.args, **kw: P.kwargs) -> bool:
     return self.storage.exists(self._get_call_hash(args, kw))

{checkpointer-2.14.8 → checkpointer-2.14.9}/checkpointer/object_hash.py

@@ -35,6 +35,8 @@ else:
 nc = nullcontext()
 stdlib = Path(sysconfig.get_paths()["stdlib"]).resolve()
 
+Buffer = bytes | bytearray | memoryview
+
 def encode_type(t: type | FunctionType) -> str:
   return f"{t.__module__}:{t.__qualname__}"
 
@@ -69,7 +71,7 @@ class ObjectHash:
   def nested_hash(self, *objs: object) -> str:
     return ObjectHash(iter=objs, tolerable=self.tolerable.value).hexdigest()
 
-  def write_bytes(self, *data: bytes, iter: Iterable[bytes] = ()) -> Self:
+  def write_bytes(self, *data: Buffer, iter: Iterable[Buffer] = ()) -> Self:
     for d in chain(data, iter):
       self.hash.update(d)
     return self

{checkpointer-2.14.8 → checkpointer-2.14.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "checkpointer"
-version = "2.14.8"
+version = "2.14.9"
 requires-python = ">=3.11"
 dependencies = []
 authors = [

{checkpointer-2.14.8 → checkpointer-2.14.9}/uv.lock

@@ -8,7 +8,7 @@ resolution-markers = [
 
 [[package]]
 name = "checkpointer"
-version = "2.14.8"
+version = "2.14.9"
 source = { editable = "." }
 
 [package.dev-dependencies]