update CI test for newer rms_jobinfo metric

Signed-off-by: Karl W Schulz <[email protected]>
AMDResearch · Jul 15, 2024 · ec3d9da · ec3d9da
1 parent 1b88114
commit ec3d9da
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/test_integration.py b/test/test_integration.py
@@ -33,8 +33,8 @@ def test_query_up(self):
 
     def test_query_slurm(self):
         prometheus = PrometheusConnect(url=self.url)
-        results = prometheus.custom_query("slurmjob_info")
-        assert len(results) >= 1, "Metric slurmjob_info not available"
+        results = prometheus.custom_query("rmsjob_info")
+        assert len(results) >= 1, "Metric rmsjob_info not available"
 
     @pytest.mark.skipif(not rocm_host, reason="requires ROCm")
     def test_query_rocm(self):
@@ -49,7 +49,7 @@ def test_query_rocm(self):
 
     def test_job(self):
         prometheus = PrometheusConnect(url=self.url)
-        query = f"slurmjob_info{{jobid=~'.+'}}[{self.time_range}]"
+        query = f"rmsjob_info{{jobid=~'.+'}}[{self.time_range}]"
         results = prometheus.custom_query(query)
 
         last_jobid = 0