From 8dbaaaa8fdae146e8635705d8cd4b3270d67f843 Mon Sep 17 00:00:00 2001 From: KPrasch Date: Fri, 12 Jan 2024 12:18:25 +0100 Subject: [PATCH] Apply suggestions from code review Co-authored-by: Derek Pierre --- nucypher/utilities/prometheus/collector.py | 4 ++-- tests/integration/utilities/test_prometheus_collectors.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/nucypher/utilities/prometheus/collector.py b/nucypher/utilities/prometheus/collector.py index 047dfb43d..ad4f51bcf 100644 --- a/nucypher/utilities/prometheus/collector.py +++ b/nucypher/utilities/prometheus/collector.py @@ -162,8 +162,8 @@ class StakingProviderMetricsCollector(BaseMetricsCollector): def initialize(self, registry: CollectorRegistry) -> None: self.metrics = { "active_stake_gauge": Gauge( - "associated_active_stake", - "Total amount of T staked (adapted NU/KEEP and liquid T)", + "active_stake", + "Total amount of T staked", registry=registry, ), "operator_confirmed_gauge": Gauge( diff --git a/tests/integration/utilities/test_prometheus_collectors.py b/tests/integration/utilities/test_prometheus_collectors.py index e224dc28f..32af865a7 100644 --- a/tests/integration/utilities/test_prometheus_collectors.py +++ b/tests/integration/utilities/test_prometheus_collectors.py @@ -129,7 +129,7 @@ def test_staking_provider_metrics_collector(test_registry, staking_providers): TACoApplicationAgent, registry=test_registry ) - active_stake = collector_registry.get_sample_value("associated_active_stake") + active_stake = collector_registry.get_sample_value("active_stake") # only floats can be stored assert active_stake == float( int(