RUN-15689 Tests fixes (#62)

run-ai · Jan 29, 2024 · fb484f1 · fb484f1
1 parent 4649803
commit fb484f1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/internal/status-exporter/app_test.go b/internal/status-exporter/app_test.go
@@ -186,6 +186,7 @@ func getTestCases() map[string]testCase {
 				"nvidia.com/gpu.count":    "1",
 				"nvidia.com/mig.strategy": "mixed",
 				"nvidia.com/gpu.product":  "Tesla P100",
+				"run.ai/fake.gpu":         "true",
 			},
 			expectedMetrics: []*dto.MetricFamily{
 				{
@@ -310,6 +311,7 @@ func getTestCases() map[string]testCase {
 				"nvidia.com/gpu.count":    "2",
 				"nvidia.com/mig.strategy": "mixed",
 				"nvidia.com/gpu.product":  "Tesla P100",
+				"run.ai/fake.gpu":         "true",
 			},
 			expectedMetrics: []*dto.MetricFamily{
 				{