Skip to content

Commit 3f60598

Browse files
committed
Fix memory_usage call in benchmarks_memory.py to pass function directly
1 parent 73d2262 commit 3f60598

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

benchmarks/benchmarks_memory.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ def main():
6868

6969
for name, func in package_funcs.items():
7070
try:
71-
mem_usage = memory_usage((func,), max_iterations=1, retval=True)
71+
mem_usage = memory_usage(func, max_iterations=1, retval=True)
7272
peak_mem = max(mem_usage[0]) # mem_usage[0] is the memory usage list
7373
wer_value = mem_usage[1] # mem_usage[1] is the return value
7474
memory_results[name] = (peak_mem, wer_value)

0 commit comments

Comments
 (0)