'Improving'

mate-academy · Sep 29, 2023 · c234964 · c234964
1 parent 5f4dbd3
commit c234964
Showing 1 changed file with 7 additions and 9 deletions.
diff --git a/app/main.py b/app/main.py
@@ -4,17 +4,15 @@
 def cache(func: Callable) -> Callable:
     cache_dict = {}
 
-    def inner(*args, **kwargs) -> None:
-        key = args
+    def inner(*args, **kwargs) -> [list, int]:
 
-        if key in cache_dict:
+        if args in cache_dict:
             print("Getting from cache")
-            return cache_dict[key]
-        else:
-            print("Calculating new result")
-            result = func(*args, **kwargs)
-            cache_dict[key] = result
+            return cache_dict[args]
 
-            return result
+        print("Calculating new result")
+        cache_dict[args] = func(*args, **kwargs)
+
+        return cache_dict[args]
 
     return inner