Memoizing previous results with lru_cache