feature: support custom hub_name in fine-tuning trainers and model evaluators for private hub testing#5781
Closed
mollyheamazon wants to merge 1 commit intoaws:masterfrom
Closed
feature: support custom hub_name in fine-tuning trainers and model evaluators for private hub testing#5781mollyheamazon wants to merge 1 commit intoaws:masterfrom
mollyheamazon wants to merge 1 commit intoaws:masterfrom
Conversation
zachgk
reviewed
Apr 21, 2026
| Can include training and validation datasets. | ||
| environment (Optional[Dict[str, str]]): | ||
| Environment variables to set in the training container. | ||
| hub_name (Optional[str]): |
Contributor
There was a problem hiding this comment.
This isn't a customer accessible feature so we should avoid having this documented where customers can see it
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Issue
Fine-tuning trainers (
SFTTrainer,DPOTrainer,RLVRTrainer,RLAIFTrainer)and model evaluators (
BenchMarkEvaluator,CustomScorerEvaluator) hardcodehub_name="SageMakerPublicHub"when fetching recipes, override parameters, andstudio tags. This blocks development and E2E testing of new recipes — a recipe
can only be exercised end-to-end after it ships to the public JumpStart hub,
delaying any PySDK fixes or support work that depends on the recipe.
Studio already uses private hubs for its own E2E suite; PySDK trainers need to
support the same pattern so test accounts can publish a pre-release recipe to a
private hub and exercise the full trainer path against it.
Description
Adds an optional
hub_nameparameter that threads through all hub-contentlookups in the trainer and evaluator flows. Defaults to
"SageMakerPublicHub"so existing callers are unaffected.
Usage
By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.