Add retries to _GetCpuUtilizationSample in CBT YCSB benchmark

PiperOrigin-RevId: 530957446
meta-introspector · May 10, 2023 · 4a98a65 · 4a98a65
1 parent 96df9d9
commit 4a98a65
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/perfkitbenchmarker/linux_benchmarks/cloud_bigtable_ycsb_benchmark.py b/perfkitbenchmarker/linux_benchmarks/cloud_bigtable_ycsb_benchmark.py
@@ -232,6 +232,7 @@ def _Install(vm: virtual_machine.VirtualMachine, bigtable: _Bigtable) -> None:
         vm.RemoteCopy(file_path, remote_path)
 
 
+@vm_util.Retry()
 def _GetCpuUtilizationSample(samples: List[sample.Sample],
                              instance_id: str) -> List[sample.Sample]:
   """Gets a list of cpu utilization samples - one per cluster per workload.