'Solution'

mate-academy · Nov 16, 2024 · b33fa21 · b33fa21
1 parent e30e510
commit b33fa21
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/app/main.py b/app/main.py
@@ -4,11 +4,13 @@
 def cache(func: Callable) -> Any:
     results = {}
 
-    def wrapper(*args, **kwargs) -> Any:
+    def wrapper(*args) -> Any:
         if args in results:
-            print(f"Getting from cache {results[args]}")
+            print("Getting from cache")
+            return results[args]
         else:
             result = func(*args)
             results[args] = result
-            print(f"Calculating new result {result}")
+            print("Calculating new result")
+            return result
     return wrapper