diff --git a/src/serve.py b/src/serve.py index 048999f..75d8c96 100644 --- a/src/serve.py +++ b/src/serve.py @@ -41,7 +41,9 @@ def make_callback(analysis_endpoint): def main(): if CONFIG.service.monitoring_enabled: tracemalloc.start() - memory_peak_metric = Gauge("pyinfra_memory_peak_per_call", "Pyinfra memory peak when processing a message in Mb") + memory_peak_metric = Gauge( + "pyinfra_memory_peak_per_call", "Pyinfra memory peak when processing a message in Mb" + ) PROM_REGISTER.register(memory_peak_metric) webserver = Process(target=run_probing_webserver, args=(set_up_probing_webserver(),)) @@ -63,7 +65,7 @@ def main(): consumer.consume_and_publish() if CONFIG.service.monitoring_enabled: _, peak = tracemalloc.get_traced_memory() - memory_peak_metric.set(peak / 10 ** 6) + memory_peak_metric.set(peak / 10**6) except Exception as err: raise ConsumerError from err