checkpointer 2.14.5__tar.gz → 2.14.6__tar.gz

{checkpointer-2.14.5 → checkpointer-2.14.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: checkpointer
-Version: 2.14.5
+Version: 2.14.6
 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.5 → checkpointer-2.14.6}/checkpointer/checkpoint.py

@@ -263,6 +263,9 @@ class CachedFunction(Generic[Fn]):
   def set(self, value, *args, **kw):
     self.storage.store(self._get_call_hash(args, kw), value)
 
+  def set_awaitable(self: CachedFunction[Callable[P, Coro[R]]], value: R, *args: P.args, **kw: P.kwargs):
+    self.set(AwaitableValue(value), *args, **kw)
+
   def __repr__(self) -> str:
     initialized = "fn_hash" in self.ident.__dict__
     fn_hash = self.ident.fn_hash[:6] if initialized else "- uninitialized"

{checkpointer-2.14.5 → checkpointer-2.14.6}/checkpointer/storages/pickle_storage.py

@@ -53,7 +53,8 @@ class PickleStorage(Storage):
     if invalidated and fn_path.exists():
       old_dirs = [path for path in fn_path.iterdir() if path.is_dir() and path != version_path]
       for path in old_dirs:
-        shutil.rmtree(path)
+        for pkl_path in path.glob("**/*.pkl"):
+          pkl_path.unlink(missing_ok=True)
       if old_dirs:
         print(f"Removed {len(old_dirs)} invalidated directories for {self.cached_fn.__qualname__}")
     if expired and self.checkpointer.expiry:

{checkpointer-2.14.5 → checkpointer-2.14.6}/checkpointer/storages/storage.py

@@ -22,7 +22,7 @@ class Storage:
     return self.checkpointer.directory / self.fn_id()
 
   def expired(self, call_hash: str) -> bool:
-    if not self.checkpointer.expiry:
+    if self.checkpointer.expiry is None:
       return False
     return self.expired_dt(self.checkpoint_date(call_hash))
 

{checkpointer-2.14.5 → checkpointer-2.14.6}/checkpointer/utils.py

@@ -121,15 +121,15 @@ def empty_dirs(path: Path) -> Iterable[Path]:
   for child in path.iterdir():
     nonempty_count += 1
     if child.is_dir():
-      for grand_child in empty_dirs(child):
-        yield grand_child
-        nonempty_count -= child == grand_child
+      for descendant in empty_dirs(child):
+        yield descendant
+        nonempty_count -= child == descendant
   if nonempty_count == 0:
     yield path
 
 def clear_directory(path: Path):
   if path.is_dir():
     for file in path.glob("**/.DS_Store"):
-      file.unlink()
+      file.unlink(missing_ok=True)
     for directory in empty_dirs(path):
       directory.rmdir()

{checkpointer-2.14.5 → checkpointer-2.14.6}/pyproject.toml

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

{checkpointer-2.14.5 → checkpointer-2.14.6}/uv.lock

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