We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7b865f5 commit 11233e1Copy full SHA for 11233e1
src/sagemaker/pytorch/estimator.py
@@ -1224,6 +1224,15 @@ def _setup_for_nova_recipe(
1224
)
1225
args["hyperparameters"]["kms_key"] = kms_key
1226
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
+
1236
# Resolve and save the final recipe
1237
self._recipe_resolve_and_save(recipe, recipe_name, args["source_dir"])
1238
0 commit comments