def enable_gc_diagnosis_and_log(diag, level="INFO"): disable_gc_diagnosis(force=True) # just in case if gc.callbacks: print("Unexpected gc.callbacks", gc.callbacks) with captured_logger("distributed.utils_perf", level=level, propagate=False) as sio: gc.disable() gc.collect() # drain any leftover from previous tests diag.enable() try: yield sio finally: diag.disable() gc.enable()
def enable_gc_diagnosis_and_log(diag): disable_gc_diagnosis(force=True) # just in case assert not gc.callbacks with captured_logger('distributed.utils_perf', level='INFO', propagate=False) as sio: gc.disable() diag.enable() try: yield sio finally: diag.disable() gc.enable()
def enable_gc_diagnosis_and_log(diag, level='INFO'): disable_gc_diagnosis(force=True) # just in case assert not gc.callbacks with captured_logger('distributed.utils_perf', level=level, propagate=False) as sio: gc.disable() gc.collect() # drain any leftover from previous tests diag.enable() try: yield sio finally: diag.disable() gc.enable()