add **kwargs

mate-academy · Sep 20, 2023 · 3e6ef07 · 3e6ef07
1 parent 6445f39
commit 3e6ef07
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/app/main.py b/app/main.py
@@ -6,9 +6,9 @@ def cache(func: Callable) -> Callable:
     result: dict = {}
 
     @wraps(func)
-    def inner(*args) -> int:
+    def inner(*args, **kwargs) -> int:
         if args not in result.keys():
-            result[args] = func(*args)
+            result[args] = func(*args, **kwargs)
             print("Calculating new result")
         else:
             print("Getting from cache")