Skip to content

Commit 11233e1

Browse files
committed
Add nova custom lambda in hyperparameter from estimator
1 parent 7b865f5 commit 11233e1

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

src/sagemaker/pytorch/estimator.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1224,6 +1224,15 @@ def _setup_for_nova_recipe(
12241224
)
12251225
args["hyperparameters"]["kms_key"] = kms_key
12261226

1227+
# Handle eval custom lambda configuration
1228+
if recipe.get("evaluation", {}):
1229+
processor = recipe.get("processor", {})
1230+
lambda_arn = processor.get("lambda_arn", "")
1231+
if lambda_arn:
1232+
args["hyperparameters"]["lambda_arn"] = lambda_arn
1233+
# Resolve and save the final recipe
1234+
self._recipe_resolve_and_save(recipe, recipe_name, args["source_dir"])
1235+
12271236
# Resolve and save the final recipe
12281237
self._recipe_resolve_and_save(recipe, recipe_name, args["source_dir"])
12291238

0 commit comments

Comments
 (0)