diff --git a/ocs_ci/ocs/cluster_load.py b/ocs_ci/ocs/cluster_load.py index 1200ef8c369..e98d7d3d19e 100644 --- a/ocs_ci/ocs/cluster_load.py +++ b/ocs_ci/ocs/cluster_load.py @@ -10,6 +10,7 @@ from range_key_dict import RangeKeyDict +from ocs_ci.utility.retry import retry from ocs_ci.utility.prometheus import PrometheusAPI from ocs_ci.utility.utils import get_trim_mean from ocs_ci.utility import templating @@ -313,6 +314,7 @@ def reach_cluster_load_percentage(self): logger.info(wrap_msg(msg)) self.target_pods_number = len(self.dc_objs) + @retry(IndexError, tries=5, delay=5, backoff=1) def get_query(self, query, mute_logs=False): """ Get query from Prometheus and parse it