Skip to content

Commit 0fd2bef

Browse files
jam-jeepintaoz-awspintaozMarta Aleszewiczrsareddy0329
authored
Release tg (#209)
* Add labels to the top level metadata (#158) Co-authored-by: pintaoz <[email protected]> * Implemented GPU Quota Allocation Feature. Co-authored-by: aleszewi <[email protected]> * Revert "Implemented GPU Quota Allocation Feature." This reverts commit 790b8f1df59494a982463aaed9e5b3f2afa44123. * Fix: Template issue - pick user defined template version (#154) * Fix: Template issue - pick user defined template version * Fix: Template issue - pick user defined template version & add topology labels in 1.1 * Fix: Template issue - pick user defined template version & add topology labels in 1.1 --------- Co-authored-by: Roja Reddy Sareddy <[email protected]> * Fix: Add __init__ to the new schema (#163) * Fix: Template issue - pick user defined template version * Fix: Template issue - pick user defined template version & add topology labels in 1.1 * Fix: Template issue - pick user defined template version & add topology labels in 1.1 * Fix: Add __init__ to load the new schema --------- Co-authored-by: Roja Reddy Sareddy <[email protected]> * Add labels and annotations to top level metadata v1.1 (#165) * Add labels to top level metadata v1.1 * Move topology labels to annotations * Update topology parameter names * Add unit test --------- Co-authored-by: pintaoz <[email protected]> * Added GPU quota allocation. Co-authored-by: aleszewi <[email protected]> * Changed neuron key to neurondevice. (#177) Co-authored-by: Marta Aleszewicz <[email protected]> * fix: Renamed memory-in-gib to memory for consistency. (#179) cr: https://code.amazon.com/reviews/CR-214599587 Co-authored-by: Marta Aleszewicz <[email protected]> * Add validation to topology labels (#178) * Add validation to topology labels * Add validation to topology labels * Add validation to topology labels --------- Co-authored-by: Roja Reddy Sareddy <[email protected]> * Add integ tests for topology annotations (#180) * Add labels to top level metadata v1.1 * Move topology labels to annotations * Update topology parameter names * Add unit test * Topology integ tests * Add invalid test case * Add empty test case --------- Co-authored-by: pintaoz <[email protected]> * Add integration tests for gpu quota allocation feature (#184) * add integration tests for gpu quota allocation feature * add valueError assertions for invalid test cases * Updating the CHANGELOG and minor version --------- Co-authored-by: pintaoz-aws <[email protected]> Co-authored-by: pintaoz <[email protected]> Co-authored-by: Marta Aleszewicz <[email protected]> Co-authored-by: rsareddy0329 <[email protected]> Co-authored-by: Roja Reddy Sareddy <[email protected]> Co-authored-by: mx26pol <[email protected]> Co-authored-by: satish Kumar <[email protected]>
1 parent d16d1b3 commit 0fd2bef

File tree

27 files changed

+2689
-233
lines changed

27 files changed

+2689
-233
lines changed

‎CHANGELOG.md

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,11 @@
11
# Changelog
22

3+
## v3.0.3 (2025-08-13)
4+
5+
### Features
6+
7+
* Task Governance feature for training jobs.
8+
39
## v3.0.2 (2025-07-31)
410

511
### Features

‎hyperpod-custom-inference-template/pyproject.toml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,4 +20,5 @@ include-package-data = true
2020

2121
[tool.setuptools.package-data]
2222
# for each versioned subpackage, include schema.json
23-
"hyperpod_custom_inference_template.v1_0" = ["schema.json"]
23+
"*" = ["schema.json"]
24+

‎hyperpod-jumpstart-inference-template/pyproject.toml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,4 +20,5 @@ include-package-data = true
2020

2121
[tool.setuptools.package-data]
2222
# for each versioned subpackage, include schema.json
23-
"hyperpod_jumpstart_inference_template.v1_0" = ["schema.json"]
23+
"*" = ["schema.json"]
24+

‎hyperpod-pytorch-job-template/CHANGELOG.md

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,9 @@
1+
## v1.1.0 (2025-08-14)
2+
3+
### Features
4+
5+
* Added parameters for task governance feature
6+
17
## v1.0.2 (2025-07-31)
28

39
### Features

‎hyperpod-pytorch-job-template/hyperpod_pytorch_job_template/registry.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,11 +10,13 @@
1010
# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
1111
# ANY KIND, either express or implied. See the License for the specific
1212
# language governing permissions and limitations under the License.
13-
from .v1_0.model import PyTorchJobConfig # Import your model
13+
from .v1_0 import model as v1_0_model # Import your model
14+
from .v1_1 import model as v1_1_model
1415
from typing import Dict, Type
1516
from pydantic import BaseModel
1617

1718
# Direct version-to-model mapping
1819
SCHEMA_REGISTRY: Dict[str, Type[BaseModel]] = {
19-
"1.0": PyTorchJobConfig,
20+
"1.0": v1_0_model.PyTorchJobConfig,
21+
"1.1": v1_1_model.PyTorchJobConfig,
2022
}

‎hyperpod-pytorch-job-template/hyperpod_pytorch_job_template/v1_0/model.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -353,6 +353,7 @@ def to_domain(self) -> Dict:
353353
result = {
354354
"name": self.job_name,
355355
"namespace": self.namespace,
356+
"labels": metadata_labels,
356357
"spec": job_kwargs,
357358
}
358359
return result

‎hyperpod-pytorch-job-template/hyperpod_pytorch_job_template/v1_1/__init__.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
from .model import PyTorchJobConfig
2+
3+
def validate(data: dict):
4+
return PyTorchJobConfig(**data)
5+
6+
7+
__all__ = ["validate", "PyTorchJobConfig"]

0 commit comments

Comments
 (0)