trainy-policy-nightly 0.1.0.dev20250921104039__py3-none-any.whl → 0.1.0.dev20251109104221__py3-none-any.whl

trainy/__init__.py

@@ -4,7 +4,7 @@ import os
 import subprocess
 
 # Replaced with the current commit when building the wheels.
-_POLICY_COMMIT_SHA = "113b8736f2bac1e39b5c51ae6e886bedf0228371"
+_POLICY_COMMIT_SHA = "14ff34c33b11d38c00a264197f0b885f67a65c45"
 
 
 def _get_git_commit():

trainy/configs/gke.yaml

@@ -30,8 +30,6 @@ kubernetes:
            {"interfaceName":"eth8","network":"vpc8"}
          ]
    spec:
-     nodeSelector:
-       cloud.google.com/gke-nodepool: h100-mega-80gb-pool
      initContainers:
        - name: tcpxo-daemon
          image: "us-docker.pkg.dev/gce-ai-infra/gpudirect-tcpxo/tcpgpudmarxd-dev:v1.0.9"

trainy/policy/gke.py

@@ -84,11 +84,11 @@ def validate_set_kueue_dws_labels_annotations(
         queue_name: str = labels.get("kueue.x-k8s.io/queue-name", DEFAULT_QUEUE)
         priority: str = labels.get("kueue.x-k8s.io/priority-class", "low-priority")
         run_duration: str = labels.get("max-run-duration-seconds", None)
-        if queue_name != DEFAULT_QUEUE:
-            raise ValueError(
-                f"{queue_name} queue was selected, "
-                f"only {DEFAULT_QUEUE} queue is permitted for hosted Trainy clusters"
-            )
+        # if queue_name != DEFAULT_QUEUE:
+        #     raise ValueError(
+        #         f"{queue_name} queue was selected, "
+        #         f"only {DEFAULT_QUEUE} queue is permitted for hosted Trainy clusters"
+        #     )
         if priority not in PRIORITY_CLASSES:
             raise ValueError(
                 f"priority `{priority}` was selected, "

{trainy_policy_nightly-0.1.0.dev20250921104039.dist-info → trainy_policy_nightly-0.1.0.dev20251109104221.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: trainy-policy-nightly
-Version: 0.1.0.dev20250921104039
+Version: 0.1.0.dev20251109104221
 Summary: Trainy Skypilot Policy
 Author: Andrew Aikawa
 Author-email: asai@berkeley.edu

{trainy_policy_nightly-0.1.0.dev20250921104039.dist-info → trainy_policy_nightly-0.1.0.dev20251109104221.dist-info}/RECORD

@@ -1,13 +1,13 @@
-trainy/__init__.py,sha256=9j2ZVKsXI3UrjSGLh59XAK9m4Gxd-XXhpoakLlIaWPg,1236
+trainy/__init__.py,sha256=JOzvZYiDSOsMOMFV3WUJWII72wYWqtturfqtMNJJTUA,1236
 trainy/config.py,sha256=MkAVzIvbS61y73uneWzZYS_FikGYZHI__DuPmRJ1lH0,322
 trainy/configs/eks.yaml,sha256=QlUFr3w7ZcTU0c0jqoGxUlWN1nVl5yE_b5IWd0ToDgs,554
-trainy/configs/gke.yaml,sha256=abBfNGoCDlIpn8blC3jd--9xGeukP6HB1vPnNlFJJzs,5049
+trainy/configs/gke.yaml,sha256=aB52IPKZhvb2ZoNn32RMkP299z3PoW2CVE-mHA5j-oE,4972
 trainy/configs/nebius.yaml,sha256=hH0W_EKIYe5EaRC9pkUHWX3P7hbM46tg0OLrc6QatJw,448
 trainy/logging.py,sha256=CMHdnnZ2b-rNDTfU7bi6jQ3AmxVv0M4sOlghnVJIqko,1186
 trainy/policy/__init__.py,sha256=DElGvZKzOqFbUs90Pi53ZpNBE3BFsalgpYMos_4tboU,180
 trainy/policy/eks.py,sha256=FBgRprnSjmp94xQvCV0tnKTgxKBk2SPNIVNFg6PPrZI,2493
-trainy/policy/gke.py,sha256=6-6PuBl_J-fuCwI5XqbQjavECnSsKVkWhHKFlz2tWOU,8163
+trainy/policy/gke.py,sha256=ClkVT6W_cWosiW1dhnJCfun_euE0YSy8B5UA-u-7f9I,8173
 trainy/policy/nebius.py,sha256=79zYL0Z6JsOomB0IF3CimSnkZfyyPvNYaJNQb0iv_Uw,3062
-trainy_policy_nightly-0.1.0.dev20250921104039.dist-info/METADATA,sha256=P05ztFZ6sI4RV0Oei3RwmvxB6I6mx7Oo_-9_9vTXnJE,1228
-trainy_policy_nightly-0.1.0.dev20250921104039.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-trainy_policy_nightly-0.1.0.dev20250921104039.dist-info/RECORD,,
+trainy_policy_nightly-0.1.0.dev20251109104221.dist-info/METADATA,sha256=BMHRfh4Eniy-yTNjDxilMBFY3TbDNXa_S8VG1F0L13A,1228
+trainy_policy_nightly-0.1.0.dev20251109104221.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+trainy_policy_nightly-0.1.0.dev20251109104221.dist-info/RECORD,,