PYMON #2 - Returning memory usage

CFMTech · Mar 16, 2020 · 6fe68ca · 6fe68ca
1 parent 2a9a0e4
commit 6fe68ca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytest_monitor/pytest_monitor.py b/pytest_monitor/pytest_monitor.py
@@ -137,7 +137,7 @@ def prof():
         m = memory_profiler.memory_usage((wrapped_function, ()), max_usage=True, retval=True)
         if isinstance(m[1], BaseException): # Do we have any outcome?
             raise m[1]
-        setattr(pyfuncitem, 'mem_usage', m)
+        setattr(pyfuncitem, 'mem_usage', m[0])
         setattr(pyfuncitem, 'monitor_results', True)
     prof()
     return True