From 46247a5cf5161e49e4f98742221c7b5569127084 Mon Sep 17 00:00:00 2001 From: Shing Chan Date: Tue, 12 Sep 2023 15:38:23 +0100 Subject: [PATCH] test(memory_profile): use params that force runs --- memory_profiler.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/memory_profiler.py b/memory_profiler.py index 28585b2..533e01a 100644 --- a/memory_profiler.py +++ b/memory_profiler.py @@ -46,14 +46,14 @@ def main(): if args.calibrate_gravity: timer.start('Calibrating gravity...') with memray.Tracker(f'{memray_dir}/calibrate_gravity.bin'): - data, info_calibrate = P.calibrate_gravity(data) + data, info_calibrate = P.calibrate_gravity(data, calib_cube=0, calib_min_samples=1) timer.stop() info.update(info_calibrate) if args.detect_nonwear: timer.start('Detecting nonwear...') with memray.Tracker(f'{memray_dir}/detect_nonwear.bin'): - data, info_nonwear = P.detect_nonwear(data) + data, info_nonwear = P.detect_nonwear(data, patience='1m') timer.stop() info.update(info_nonwear) @@ -69,7 +69,7 @@ def main(): if fname.endswith('.bin'): subprocess.run(['python3', '-m', 'memray', 'flamegraph', f'{memray_dir}/{fname}'], check=True) - print(data.head()) + print(data) # Pretty print info for k, v in info.items():