trainy-policy-nightly 0.1.0.dev20250730105055__py3-none-any.whl → 0.1.0.dev20251106104717__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of trainy-policy-nightly might be problematic. Click here for more details.

Files changed (6) hide show
  1. trainy/__init__.py +1 -1
  2. trainy/configs/gke.yaml +0 -2
  3. trainy/policy/gke.py +6 -6
  4. {trainy_policy_nightly-0.1.0.dev20250730105055.dist-info → trainy_policy_nightly-0.1.0.dev20251106104717.dist-info}/METADATA +1 -1
  5. {trainy_policy_nightly-0.1.0.dev20250730105055.dist-info → trainy_policy_nightly-0.1.0.dev20251106104717.dist-info}/RECORD +6 -6
  6. {trainy_policy_nightly-0.1.0.dev20250730105055.dist-info → trainy_policy_nightly-0.1.0.dev20251106104717.dist-info}/WHEEL +0 -0
trainy/__init__.py CHANGED
@@ -4,7 +4,7 @@ import os
4
4
  import subprocess
5
5
 
6
6
  # Replaced with the current commit when building the wheels.
7
- _POLICY_COMMIT_SHA = "5a5eb888dd9bed30347a7b6153198835c149960f"
7
+ _POLICY_COMMIT_SHA = "14ff34c33b11d38c00a264197f0b885f67a65c45"
8
8
 
9
9
 
10
10
  def _get_git_commit():
trainy/configs/gke.yaml CHANGED
@@ -30,8 +30,6 @@ kubernetes:
30
30
  {"interfaceName":"eth8","network":"vpc8"}
31
31
  ]
32
32
  spec:
33
- nodeSelector:
34
- cloud.google.com/gke-nodepool: h100-mega-80gb-pool
35
33
  initContainers:
36
34
  - name: tcpxo-daemon
37
35
  image: "us-docker.pkg.dev/gce-ai-infra/gpudirect-tcpxo/tcpgpudmarxd-dev:v1.0.9"
trainy/policy/gke.py CHANGED
@@ -20,7 +20,7 @@ logger = get_logger(__file__)
20
20
 
21
21
  DEFAULT_QUEUE = "user-queue"
22
22
  PRIORITY_CLASSES = ["low-priority", "high-priority"]
23
- ALLOWED_GPUS = ["H100-MEGA-80GB", "A100-80GB", "A100"]
23
+ ALLOWED_GPUS = ["H100", "H100-MEGA-80GB", "A100-80GB", "A100"]
24
24
 
25
25
 
26
26
  def set_tcpxo_config(user_request: sky.UserRequest) -> sky.MutatedUserRequest:
@@ -84,11 +84,11 @@ def validate_set_kueue_dws_labels_annotations(
84
84
  queue_name: str = labels.get("kueue.x-k8s.io/queue-name", DEFAULT_QUEUE)
85
85
  priority: str = labels.get("kueue.x-k8s.io/priority-class", "low-priority")
86
86
  run_duration: str = labels.get("max-run-duration-seconds", None)
87
- if queue_name != DEFAULT_QUEUE:
88
- raise ValueError(
89
- f"{queue_name} queue was selected, "
90
- f"only {DEFAULT_QUEUE} queue is permitted for hosted Trainy clusters"
91
- )
87
+ # if queue_name != DEFAULT_QUEUE:
88
+ # raise ValueError(
89
+ # f"{queue_name} queue was selected, "
90
+ # f"only {DEFAULT_QUEUE} queue is permitted for hosted Trainy clusters"
91
+ # )
92
92
  if priority not in PRIORITY_CLASSES:
93
93
  raise ValueError(
94
94
  f"priority `{priority}` was selected, "
{trainy_policy_nightly-0.1.0.dev20250730105055.dist-info → trainy_policy_nightly-0.1.0.dev20251106104717.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: trainy-policy-nightly
3
- Version: 0.1.0.dev20250730105055
3
+ Version: 0.1.0.dev20251106104717
4
4
  Summary: Trainy Skypilot Policy
5
5
  Author: Andrew Aikawa
6
6
  Author-email: asai@berkeley.edu
{trainy_policy_nightly-0.1.0.dev20250730105055.dist-info → trainy_policy_nightly-0.1.0.dev20251106104717.dist-info}/RECORD RENAMED
@@ -1,13 +1,13 @@
1
- trainy/__init__.py,sha256=vTajkc9lnsg5xXk48tZKOjqXZVLpqlwt7JlQrPM41bE,1236
1
+ trainy/__init__.py,sha256=JOzvZYiDSOsMOMFV3WUJWII72wYWqtturfqtMNJJTUA,1236
2
2
  trainy/config.py,sha256=MkAVzIvbS61y73uneWzZYS_FikGYZHI__DuPmRJ1lH0,322
3
3
  trainy/configs/eks.yaml,sha256=QlUFr3w7ZcTU0c0jqoGxUlWN1nVl5yE_b5IWd0ToDgs,554
4
- trainy/configs/gke.yaml,sha256=abBfNGoCDlIpn8blC3jd--9xGeukP6HB1vPnNlFJJzs,5049
4
+ trainy/configs/gke.yaml,sha256=aB52IPKZhvb2ZoNn32RMkP299z3PoW2CVE-mHA5j-oE,4972
5
5
  trainy/configs/nebius.yaml,sha256=hH0W_EKIYe5EaRC9pkUHWX3P7hbM46tg0OLrc6QatJw,448
6
6
  trainy/logging.py,sha256=CMHdnnZ2b-rNDTfU7bi6jQ3AmxVv0M4sOlghnVJIqko,1186
7
7
  trainy/policy/__init__.py,sha256=DElGvZKzOqFbUs90Pi53ZpNBE3BFsalgpYMos_4tboU,180
8
8
  trainy/policy/eks.py,sha256=FBgRprnSjmp94xQvCV0tnKTgxKBk2SPNIVNFg6PPrZI,2493
9
- trainy/policy/gke.py,sha256=QNm39QDp0Q-8r_TQi1EmFTU7mgp8b-Wwd6wtfmD61G4,8155
9
+ trainy/policy/gke.py,sha256=ClkVT6W_cWosiW1dhnJCfun_euE0YSy8B5UA-u-7f9I,8173
10
10
  trainy/policy/nebius.py,sha256=79zYL0Z6JsOomB0IF3CimSnkZfyyPvNYaJNQb0iv_Uw,3062
11
- trainy_policy_nightly-0.1.0.dev20250730105055.dist-info/METADATA,sha256=Fy_wDxG2w7_UjUlOp7s9EMRow26Ix7SaL1oDsM8u6sA,1228
12
- trainy_policy_nightly-0.1.0.dev20250730105055.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
13
- trainy_policy_nightly-0.1.0.dev20250730105055.dist-info/RECORD,,
11
+ trainy_policy_nightly-0.1.0.dev20251106104717.dist-info/METADATA,sha256=Vb0gfgPzMlpDLsdNrxXwCb0NJUYA2z4ALSLhAMtXMrM,1228
12
+ trainy_policy_nightly-0.1.0.dev20251106104717.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
13
+ trainy_policy_nightly-0.1.0.dev20251106104717.dist-info/RECORD,,