Skip to content

Commit a2c6bf4

Browse files
Update decorators_bilal_ayakdas.py
1 parent 836b043 commit a2c6bf4

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

Week04/decorators_bilal_ayakdas.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,12 +15,13 @@ def performance(fn):
1515

1616
def calculate_perf(*args, **kwargs):
1717
begin_time = time()
18-
memory_usage = getsizeof(fn(*args,**kwargs))
18+
result = fn(*args,**kwargs)
19+
memory_usage = getsizeof(result)
1920
end_time = time()
2021

2122
setattr(performance,"counter",getattr(performance,"counter") + 1)
2223
setattr(performance,"total_time",getattr(performance,"total_time") + (end_time - begin_time))
2324
setattr(performance,"total_mem",getattr(performance,"total_mem") + memory_usage)
2425

25-
return fn(*args,**kwargs)
26+
return result
2627
return calculate_perf

0 commit comments

Comments
 (0)