Try to fix Valentine's remarks ver.2

mate-academy · Dec 3, 2024 · f7828d6 · f7828d6
1 parent 8866462
commit f7828d6
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/app/main.py b/app/main.py
@@ -1,17 +1,16 @@
-from typing import Callable
+from typing import Callable, Any
 
 
 def cache(func: Callable) -> Callable:
     cached_data = {}
 
-    def wrapper(*args) -> Callable:
-
+    def wrapper(*args) -> Any:
         if args in cached_data:
             print("Getting from cache")
-            return cached_data[args]
+
         else:
             print("Calculating new result")
             cached_data[args] = func(*args)
-            return cached_data[args]
+        return cached_data[args]
 
     return wrapper