diff --git a/pyinfra/server/server.py b/pyinfra/server/server.py index 65396f7..aac505a 100644 --- a/pyinfra/server/server.py +++ b/pyinfra/server/server.py @@ -53,13 +53,14 @@ def __set_up_processing_server_impl(operation2function: Dict[str, QueuedStreamFu } def make_summary_instance(op): + op = op.replace("_pickup", "") return Summary(f"redactmanager_{op}_seconds", f"Time spent on {op}.", registry=registry) submit_operation2processor = {submit_suffix(op): prc for op, prc in operation2processor.items()} pickup_operation2processor = {pickup_suffix(op): prc for op, prc in operation2processor.items()} operation2processor = merge(submit_operation2processor, pickup_operation2processor) - operation2metric = {op: make_summary_instance(op) for op in operation2processor} + operation2metric = {op: make_summary_instance(op) for op in pickup_operation2processor} @app.route("/ready", methods=["GET"]) def ready():