black is back

This commit is contained in:
Julius Unverfehrt 2022-03-16 08:59:45 +01:00
parent 5950799e03
commit 9f3c884c75

View File

@ -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