mirror of
https://github.com/hanzo-ml/trainer.git
synced 2026-09-24 00:31:34 +00:00
Krishna GuptaandGitHub
ccaf20bc20
* Add XGBoost runtime plugin scaffold and register in framework Signed-off-by: Krishna-kg732 <[email protected]> Signed-off-by: krishna-kg732 <[email protected]> * regenerated assets Signed-off-by: Krishna-kg732 <[email protected]> Signed-off-by: krishna-kg732 <[email protected]> * Add XGBoost plugin unit tests and update framework test registry Signed-off-by: Krishna-kg732 <[email protected]> Signed-off-by: krishna-kg732 <[email protected]> * added :Runtime image , unit tests for :(Resources are not set, Resources are set in TrainJob) ,using the ContainerTrainerPort instead of default port Signed-off-by: krishna-kg732 <[email protected]> * minor changes Signed-off-by: krishna-kg732 <[email protected]> * added plainML fallback(JAX+XGBOOST) Signed-off-by: krishna-kg732 <[email protected]> * test: add XGBoost runtime resource resolution unit tests - Add test case for resources set in Runtime only - Add test case for resources set in both Runtime and TrainJob - Add xgboost_distributed.yaml ClusterTrainingRuntime manifest - Update kustomization.yaml with xgboost-runtime image Addresses feedback from PR #3200 for complete resource resolution test coverage. Signed-off-by: krishna-kg732 <[email protected]> * added autogenerat3ed files Signed-off-by: krishna-kg732 <[email protected]> * CI fixes Signed-off-by: krishna-kg732 <[email protected]> * chore: fix ci lint errors Signed-off-by: krishna-kg732 <[email protected]> * Add XGBoost integration and E2E tests, fix Validate guard clause and nil safety Signed-off-by: krishna-kg732 <[email protected]> * trigger ci Signed-off-by: krishna-kg732 <[email protected]> * fix: Add XGBoost support to test wrapper, CEL validation, and CI publish workflow - Add XGBoostPolicy() method to MLPolicySourceWrapper for integration tests - Add has(self.xgboost) to MLPolicy CEL mutual-exclusivity validation rule - Regenerate CRD manifests with updated CEL rule - Add xgboost-runtime to build-and-push-images CI workflow - Add XGBoost image build step to E2E setup script (commented out due to Kind bug) Signed-off-by: krishna-kg732 <[email protected]> * docs: Add XGBoost distributed training example notebook Add example notebook demonstrating distributed XGBoost training using the Kubeflow Trainer SDK and the xgboost-distributed ClusterTrainingRuntime. The notebook uses the modern XGBoost Collective API with RabitTracker for distributed coordination and synthetic classification data. Signed-off-by: krishna-kg732 <[email protected]> * test: Skip XGBoost E2E test until runtime image is published Signed-off-by: krishna-kg732 <[email protected]> * chore : address review comments Signed-off-by: krishna-kg732 <[email protected]> * chore:resolve merge conflicts Signed-off-by: krishna-kg732 <[email protected]> --------- Signed-off-by: Krishna-kg732 <[email protected]> Signed-off-by: krishna-kg732 <[email protected]>