Commit 7ae43c8
Fix HyperparameterTuner to include ModelTrainer internal channels (#5516)
* Fix HyperparameterTuner to include ModelTrainer internal channels
HyperparameterTuner._build_training_job_definition() was only copying
user-provided input channels, missing ModelTrainer's internal channels
(code, sm_drivers). This caused tuning jobs to fail when using custom
training scripts.
- Add logic to include ModelTrainer's input_data_config channels
- Refactor test_tuner.py with factory functions to reduce duplication
- Add test for internal channels inclusion
Fixes #5508
* fix formatting
* Fix static_hyperparameters initialization and reference
---------
Co-authored-by: adishaa <adishaa@amazon.com>
Co-authored-by: Sharma <adishaa@14147d9496bd.ant.amazon.com>1 parent ea3ea86 commit 7ae43c8
File tree
2 files changed
+288
-174
lines changed- sagemaker-train
- src/sagemaker/train
- tests/unit/train
2 files changed
+288
-174
lines changed
0 commit comments