From a893a5f571adaf68ac2ffa26746e6ba2dab4a0bd Mon Sep 17 00:00:00 2001 From: gshaibi Date: Tue, 16 Jul 2024 00:03:34 +0300 Subject: [PATCH] . --- internal/status-exporter/app_test.go | 28 ++++++++++++---------------- 1 file changed, 12 insertions(+), 16 deletions(-) diff --git a/internal/status-exporter/app_test.go b/internal/status-exporter/app_test.go index 5df3cbd..7c5039f 100644 --- a/internal/status-exporter/app_test.go +++ b/internal/status-exporter/app_test.go @@ -182,14 +182,12 @@ func getTestCases() map[string]testCase { }, }, expectedLabels: map[string]string{ - "nvidia.com/gpu.present": "true", - "nvidia.com/gpu.memory": "20000", - "nvidia.com/gpu.count": "1", - "nvidia.com/mig.strategy": "mixed", - "nvidia.com/gpu.product": "Tesla P100", - "run.ai/fake.gpu": "true", - "nvidia.com/gpu.deploy.dcgm-exporter": "true", - "nvidia.com/gpu.deploy.device-plugin": "true", + "nvidia.com/gpu.present": "true", + "nvidia.com/gpu.memory": "20000", + "nvidia.com/gpu.count": "1", + "nvidia.com/mig.strategy": "mixed", + "nvidia.com/gpu.product": "Tesla P100", + "run.ai/fake.gpu": "true", }, expectedMetrics: []*dto.MetricFamily{ { @@ -309,14 +307,12 @@ func getTestCases() map[string]testCase { }, }, expectedLabels: map[string]string{ - "nvidia.com/gpu.present": "true", - "nvidia.com/gpu.memory": "20000", - "nvidia.com/gpu.count": "2", - "nvidia.com/mig.strategy": "mixed", - "nvidia.com/gpu.product": "Tesla P100", - "run.ai/fake.gpu": "true", - "nvidia.com/gpu.deploy.dcgm-exporter": "true", - "nvidia.com/gpu.deploy.device-plugin": "true", + "nvidia.com/gpu.present": "true", + "nvidia.com/gpu.memory": "20000", + "nvidia.com/gpu.count": "2", + "nvidia.com/mig.strategy": "mixed", + "nvidia.com/gpu.product": "Tesla P100", + "run.ai/fake.gpu": "true", }, expectedMetrics: []*dto.MetricFamily{ {