Fix

Signed-off-by: Tyler Gu <[email protected]>
xlab-uiuc · Nov 13, 2023 · 6dbd591 · 6dbd591
1 parent f738e09
commit 6dbd591
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/performance_measurement/check_utils.py b/performance_measurement/check_utils.py
@@ -51,8 +51,9 @@ def check_tolerations(
         desired_tolerations: List[dict],
         sts_object: k8s_models.V1StatefulSet) -> bool:
     # check if tolerations match
-    if desired_tolerations is None:
-        return sts_object["spec"]["template"]["spec"]["tolerations"] is None
+    if desired_tolerations is None or len(desired_tolerations) == 0:
+        return sts_object["spec"]["template"]["spec"]["tolerations"] is None or len(
+            sts_object["spec"]["template"]["spec"]["tolerations"]) == 0
     toleration_matched = True
     for toleration in desired_tolerations:
         if sts_object["spec"]["template"]["spec"]["tolerations"] is None: