Skip to content

Commit 728a51a

Browse files
author
Mohamed Zeidan
committed
updated sagemaker-core, boto
1 parent 3646f22 commit 728a51a

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

pyproject.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ classifiers = [
3232
]
3333
dependencies = [
3434
"attrs>=24,<26",
35-
"boto3>=1.39.5,<2.0",
35+
"boto3>=1.42.2,<2.0",
3636
"cloudpickle>=2.2.1",
3737
"docker",
3838
"fastapi",
@@ -49,7 +49,7 @@ dependencies = [
4949
"psutil",
5050
"PyYAML>=6.0.1",
5151
"requests",
52-
"sagemaker-core>=1.0.17,<2.0.0",
52+
"sagemaker-core>=1.0.71,<2.0.0",
5353
"schema",
5454
"smdebug_rulesconfig==1.0.1",
5555
"tblib>=1.7.0,<4",

src/sagemaker/modules/train/sm_recipes/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -351,7 +351,7 @@ def _get_args_from_nova_recipe(
351351
reward_lambda_arn = run_config.get("reward_lambda_arn", "")
352352
if reward_lambda_arn:
353353
args["hyperparameters"]["reward_lambda_arn"] = reward_lambda_arn
354-
354+
355355
_register_custom_resolvers()
356356

357357
# Resolve Final Recipe

tests/unit/test_pytorch_nova.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -909,4 +909,4 @@ def test_setup_for_nova_recipe_without_reward_lambda(mock_resolve_save, sagemake
909909
assert pytorch.is_nova_or_eval_recipe is True
910910

911911
# Verify that reward_lambda_arn hyperparameter was not set
912-
assert "reward_lambda_arn" not in pytorch._hyperparameters
912+
assert "reward_lambda_arn" not in pytorch._hyperparameters

0 commit comments

Comments
 (0)