ipynb cache

ipynb_cache.py

from typing import Dict, List, Any
import inspect

Result = Any
global_run_cache: Dict[int, Result] = {}

def cached_function(fn):
    def wrapper(*args, **kwargs):
        global global_run_cache
        consty_types = [str, int, float]
        id_of = lambda v: v if type(v) in consty_types else id(v) 
        runHash = hash(tuple([id_of(arg) for arg in args] + [(key, id_of(arg)) for key, arg in kwargs] + [inspect.getsource(fn)]))
        if runHash not in global_run_cache:
            global_run_cache[runHash] = fn(*args, **kwargs)
        return global_run_cache[runHash]
    return wrapper


# Usage:
import time

@cached_function
def some_function(params):
    time.sleep(params)
    return params

print(some_function(5))