checkpointer 2.14.1__tar.gz → 2.14.2__tar.gz

{checkpointer-2.14.1 → checkpointer-2.14.2}/PKG-INFO

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

@@ -241,6 +241,16 @@ class CachedFunction(Generic[Fn]):
     except Exception as ex:
       raise CheckpointError("Could not load checkpoint") from ex
 
+  @overload
+  def get_or(self: Callable[P, Coro[R]], default: R, *args: P.args, **kw: P.kwargs) -> R: ...
+  @overload
+  def get_or(self: Callable[P, R], default: R, *args: P.args, **kw: P.kwargs) -> R: ...
+  def get_or(self, default, *args, **kw):
+    try:
+      return self.get(*args, **kw)  # type: ignore
+    except CheckpointError:
+      return default
+
   @overload
   def set(self: Callable[P, Coro[R]], value: AwaitableValue[R], *args: P.args, **kw: P.kwargs): ...
   @overload

{checkpointer-2.14.1 → checkpointer-2.14.2}/checkpointer/storages/pickle_storage.py

@@ -2,22 +2,12 @@ import pickle
 import shutil
 from datetime import datetime
 from pathlib import Path
+from ..utils import clear_directory
 from .storage import Storage
 
 def filedate(path: Path) -> datetime:
   return datetime.fromtimestamp(path.stat().st_mtime)
 
-def empty_dirs(path: Path):
-  nonempty_count = 0
-  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
-  if nonempty_count == 0:
-    yield path
-
 class PickleStorage(Storage):
   def get_path(self, call_hash: str):
     return self.fn_dir() / f"{call_hash[:2]}/{call_hash[2:]}.pkl"
@@ -58,11 +48,7 @@ class PickleStorage(Storage):
           count += 1
           pkl_path.unlink(missing_ok=True)
       print(f"Removed {count} expired checkpoints for {self.cached_fn.__qualname__}")
-    if fn_path.exists():
-      for file in fn_path.glob("**/.DS_Store"):
-        file.unlink()
-      for directory in empty_dirs(fn_path):
-        directory.rmdir()
+    clear_directory(fn_path)
 
   def clear(self):
     fn_path = self.fn_dir().parent

{checkpointer-2.14.1 → checkpointer-2.14.2}/checkpointer/utils.py

@@ -112,3 +112,21 @@ class ContextVar(Generic[T]):
       yield
     finally:
       self.value = old
+
+def empty_dirs(path: Path) -> Iterable[Path]:
+  nonempty_count = 0
+  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
+  if nonempty_count == 0:
+    yield path
+
+def clear_directory(path: Path):
+  if path.is_dir():
+    for file in path.glob("**/.DS_Store"):
+      file.unlink()
+    for directory in empty_dirs(path):
+      directory.rmdir()

{checkpointer-2.14.1 → checkpointer-2.14.2}/pyproject.toml

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

{checkpointer-2.14.1 → checkpointer-2.14.2}/uv.lock

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