Performance decorator

Refactor performance decorator to simplify memory tracking and timing.

Author: BugraFiloglu

Week04/decorators_ilyas_bugra_filoglu.py

@@ -8,31 +8,24 @@
 def performance(fn):
     @wraps(fn)
     def wrapper(*args, **kwargs):
-
+      
         performance.counter += 1
-
-        current_before, _ = tracemalloc.get_traced_memory()
+      
         tracemalloc.reset_peak()
-
         start_time = time.perf_counter()
+        result = fn(*args, **kwargs)
+      
+        end_time = time.perf_counter()
+      
+        performance.total_time += end_time - start_time
+      
+        current_memory, peak_memory = tracemalloc.get_traced_memory()
+        performance.total_mem += peak_memory
 
-        try:
-            result = fn(*args, **kwargs)
-            return result
-
-        finally:
-            end_time = time.perf_counter()
-
-            performance.total_time += end_time - start_time
-
-            current_memory, peak_memory = tracemalloc.get_traced_memory()
-
-            performance.total_mem += max(
-                0, peak_memory - current_before
-            )
+        return result
 
     return wrapper
-
+  
 
 performance.counter = 0
 performance.total_time = 0.0