checkpointer 2.14.5__py3-none-any.whl → 2.14.6__py3-none-any.whl

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/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/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/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.dist-info → checkpointer-2.14.6.dist-info}/METADATA

@@ -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.dist-info → checkpointer-2.14.6.dist-info}/RECORD

@@ -1,18 +1,18 @@
 checkpointer/__init__.py,sha256=lprhgo4BcVbr4RczWdkNIWroVX_9M-akZcMbxOAWHf0,981
-checkpointer/checkpoint.py,sha256=Vw0uKPBtp01Rit-ydE28TrMO3t3HQ2dkT0uJT2gtyms,10509
+checkpointer/checkpoint.py,sha256=BG2LkBbnCxWAzSTCNdYFJElef7LIFu29MvZqRp0RFrc,10665
 checkpointer/fn_ident.py,sha256=Z2HtkoG8n0ddqNzjejnKW6Lo-ID2unKQZa1tOpRGujs,6911
 checkpointer/fn_string.py,sha256=YldjAU91cwiZNwuE_AQN_DMiQCPAhGiiC3lPi0uvM0g,2579
 checkpointer/import_mappings.py,sha256=ESqWvZTzYAmaVnJ6NulUvn3_8CInOOPmEKUXO2WD_WA,1794
 checkpointer/object_hash.py,sha256=_VgyTEoe3H6268KBAOFZxSgWHb6otgG-3NzhHyehxe4,8595
 checkpointer/print_checkpoint.py,sha256=uUQ493fJCaB4nhp4Ox60govSCiBTIPbBX15zt2QiRGo,1356
 checkpointer/types.py,sha256=GFqbGACdDxzQX3bb2LmF9UxQVWOEisGvdtobnqCBAOA,1129
-checkpointer/utils.py,sha256=BAv_qvcanbYqgr7cCOCKNiG0_rGbNbLomBbpvoYlltc,3553
+checkpointer/utils.py,sha256=WGfVeHeZua010YXMuDPwAYfyQyB6k2bm0n0nbZdJ28Q,3565
 checkpointer/storages/__init__.py,sha256=p-r4YrPXn505_S3qLrSXHSlsEtb13w_DFnCt9IiUomk,296
 checkpointer/storages/memory_storage.py,sha256=M5O3gRMHBiG3CgMcFdA6-iDGvfafKqTi9nN3u8sETCg,1801
-checkpointer/storages/pickle_storage.py,sha256=3_it1YQL6y2WQwnOHT_UPZ1BQkVJZvuR9txcuiuPwYQ,2242
-checkpointer/storages/storage.py,sha256=0dwxQvg0HXlgU93FRXrpZnF_Qb6-KE0WyWhS_GJmmm0,1366
-checkpointer-2.14.5.dist-info/METADATA,sha256=vNYVwNi98WgyKujBVSLhnkWqsx3VMo7KHhgOiWe-i9w,11215
-checkpointer-2.14.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-checkpointer-2.14.5.dist-info/licenses/ATTRIBUTION.md,sha256=WF6L7-sD4s9t9ytVJOhjhpDoZ6TrWpqE3_bMdDIeJxI,1078
-checkpointer-2.14.5.dist-info/licenses/LICENSE,sha256=drXs6vIb7uW49r70UuMz2A1VtOCl626kiTbcmrar1Xo,1072
-checkpointer-2.14.5.dist-info/RECORD,,
+checkpointer/storages/pickle_storage.py,sha256=L6E7cugbn-XGMMlcHCFBJ-rBLbmtzbiPwRrlgfRd7yc,2304
+checkpointer/storages/storage.py,sha256=5v_9Mp8__oqPnyrGsUEiT7jU4eMk9L4ORHDZeQw-TU4,1370
+checkpointer-2.14.6.dist-info/METADATA,sha256=q-QYZcNIAkQ-3AcnIuKaXtifxwlIqYPU3vy86gTFGBQ,11215
+checkpointer-2.14.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+checkpointer-2.14.6.dist-info/licenses/ATTRIBUTION.md,sha256=WF6L7-sD4s9t9ytVJOhjhpDoZ6TrWpqE3_bMdDIeJxI,1078
+checkpointer-2.14.6.dist-info/licenses/LICENSE,sha256=drXs6vIb7uW49r70UuMz2A1VtOCl626kiTbcmrar1Xo,1072
+checkpointer-2.14.6.dist-info/RECORD,,