fix else and add type annotation Any

mate-academy · Sep 30, 2023 · f6f87c4 · f6f87c4
1 parent ea3930a
commit f6f87c4
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/app/main.py b/app/main.py
@@ -1,15 +1,15 @@
-from typing import Callable
+from typing import Callable, Any
 
 
 def cache(func: Callable) -> Callable:
     cache_dict = {}
 
-    def wrapper(*args: Callable) -> Callable:
+    def wrapper(*args) -> Any:
         if args not in cache_dict:
             cache_dict[args] = func(*args)
             print("Calculating new result")
-        else:
-            print("Getting from cache")
+            return cache_dict[args]
+        print("Getting from cache")
         return cache_dict[args]
 
     return wrapper