mirror of https://github.com/hanzo-ml/trainer.git synced 2026-09-24 00:31:34 +00:00

Files

T

Krish GargandGitHub 4f8920c902 feat(api): Add terminationGracePeriodSeconds to PodSpecPatch in TrainJob (#3324 )

Adds terminationGracePeriodSeconds field to PodSpecPatch so users can
configure pod termination grace period per TrainJob via RuntimePatches.

This is needed for distributed training with PyTorch Elastic (torchrun)
where large models (70B+ parameters) require more than the default 30s
to complete JIT checkpointing before SIGKILL on node drain or TrainJob
pause.

No changes to merge logic in trainingruntime.go are required since the
existing StrategicMergePatch applied at batchv1.JobTemplateSpec level
already handles this field automatically.

Closes #3285

Signed-off-by: krishdef7 <[email protected]>

2026-05-26 17:36:09 +00:00

kubeflow_trainer_api

feat(api): Add terminationGracePeriodSeconds to PodSpecPatch in TrainJob (#3324 )

2026-05-26 17:36:09 +00:00

pyproject.toml

chore: Remove tool.hatch.build.targets.wheel from pyproject (#2803 )

2025-08-28 15:36:22 +00:00

README.md

Move generated Python models into kubeflow_trainer_api package (#2632 )

2025-05-07 18:14:39 +00:00

requirements-lock.txt

feat(ci): add Python dependency scanning to OSV-Scanner workflow (#3530 )

2026-05-20 13:34:59 +00:00

README.md

Kubeflow Trainer API

Python package containing Kubeflow Trainer API models.