Fix intel gpu stats (#4581)

* Fix return code for timeout call

* Rewrit intel gpu logic

* Fix test and output
This commit is contained in:
Nicolas Mowen
2022-12-03 09:28:01 -07:00
committed by GitHub
parent 94a2b6bb88
commit 007fa75294
2 changed files with 30 additions and 18 deletions

View File

@@ -35,11 +35,11 @@ class TestGpuStats(unittest.TestCase):
@patch("subprocess.run")
def test_intel_gpu_stats(self, sp):
process = MagicMock()
process.returncode = 0
process.returncode = 124
process.stdout = self.intel_results
sp.return_value = process
intel_stats = get_intel_gpu_stats()
assert intel_stats == {
"gpu": "10.73 %",
"gpu": "1.34 %",
"mem": "- %",
}