diff --git a/pyinfra/server/server.py b/pyinfra/server/server.py index 2054ebb..65396f7 100644 --- a/pyinfra/server/server.py +++ b/pyinfra/server/server.py @@ -59,7 +59,7 @@ def __set_up_processing_server_impl(operation2function: Dict[str, QueuedStreamFu pickup_operation2processor = {pickup_suffix(op): prc for op, prc in operation2processor.items()} operation2processor = merge(submit_operation2processor, pickup_operation2processor) - pickup_operation2metric = {op: make_summary_instance(op) for op in operation2processor} + operation2metric = {op: make_summary_instance(op) for op in operation2processor} @app.route("/ready", methods=["GET"]) def ready(): @@ -83,7 +83,7 @@ def __set_up_processing_server_impl(operation2function: Dict[str, QueuedStreamFu @app.route("/", methods=["GET"]) def pickup(operation): - with pickup_operation2metric[operation.replace("_pickup", "")].time(): + with operation2metric[operation.replace("_pickup", "")].time(): return operation2processor[operation].pop() return app