tinygrad/tinygrad
2
0
Fork 3
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-08-29 18:36:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixup typing for @diskcache [pr] (#10529)

Browse Source
This commit is contained in:
qazal
2025-05-27 15:30:49 +03:00
committed by GitHub
parent 41e3d07d7f
commit c03e9c8995
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/helpers.py
+2 -2
View File
@@ -220,8 +220,8 @@ def diskcache_put(table:str, key:Union[dict, str, int], val:Any, prepickled=Fals
cur.close()
return val
def diskcache(func):
def wrapper(*args, **kwargs) -> bytes:
def diskcache(func:Callable[..., T]):
def wrapper(*args, **kwargs) -> T:
table, key = f"cache_{func.__name__}", hashlib.sha256(pickle.dumps((args, kwargs))).hexdigest()
if (ret:=diskcache_get(table, key)) is not None: return ret
return diskcache_put(table, key, func(*args, **kwargs))