review fixes

Signed-off-by: Piotr Pawłowski <[email protected]>
AI-Hypercomputer · Jan 14, 2025 · ad39218 · ad39218
1 parent 4c861a3
commit ad39218
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/xpk/core/kueue.py b/src/xpk/core/kueue.py
@@ -215,10 +215,10 @@ def install_kueue_on_cluster(args) -> int:
 
   err_code, kueue_version_installed = get_kueue_version(args)
   if err_code == 0:
-    if Version(kueue_version_installed) <= Version('v0.8.1') and Version(
+    if Version(kueue_version_installed) < Version('v0.9.0') and Version(
         KUEUE_VERSION
-    ) >= Version('v0.9.1'):
-      xpk_print('Upgrading kueue on cluster from version < 0.9.1.')
+    ) >= Version('v0.9.0'):
+      xpk_print('Upgrading kueue on cluster from version < 0.9.0.')
       upgrade_code = delete_multikueueclusters_definitions(args)
       if upgrade_code != 0:
         return upgrade_code