Skip to content

Commit 03b2c9c

Browse files
committed
chore: address flake8 errors
1 parent 4656cb8 commit 03b2c9c

File tree

3 files changed

+16
-8
lines changed

3 files changed

+16
-8
lines changed

src/sagemaker/jumpstart/factory/estimator.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -612,7 +612,8 @@ def _add_env_to_kwargs(
612612
)
613613
if model_specs.is_gated_model():
614614
raise ValueError(
615-
f"Need to define ‘accept_eula'='true' within Environment. {get_eula_message(model_specs, kwargs.region)}"
615+
"Need to define ‘accept_eula'='true' within Environment. "
616+
f"{get_eula_message(model_specs, kwargs.region)}"
616617
)
617618

618619
return kwargs

tests/unit/sagemaker/jumpstart/constants.py

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,8 @@
2929
},
3030
"hosting_artifact_key": "huggingface-llm/huggingface-llm-gemma-7b-instruct/artifacts/inference/v1.0.0/",
3131
"hosting_script_key": "source-directory-tarballs/huggingface/inference/llm/v1.0.1/sourcedir.tar.gz",
32-
"hosting_prepacked_artifact_key": "huggingface-llm/huggingface-llm-gemma-7b-instruct/artifacts/inference-prepack/v1.0.0/",
32+
"hosting_prepacked_artifact_key": "huggingface-llm/huggingface-llm-gemma-7b-i"
33+
"nstruct/artifacts/inference-prepack/v1.0.0/",
3334
"hosting_prepacked_artifact_version": "1.0.0",
3435
"hosting_use_script_uri": False,
3536
"hosting_eula_key": "fmhMetadata/terms/gemmaTerms.txt",
@@ -417,7 +418,8 @@
417418
},
418419
],
419420
"training_script_key": "source-directory-tarballs/huggingface/transfer_learning/llm/v1.1.1/sourcedir.tar.gz",
420-
"training_prepacked_script_key": "source-directory-tarballs/huggingface/transfer_learning/llm/prepack/v1.1.1/sourcedir.tar.gz",
421+
"training_prepacked_script_key": "source-directory-tarballs/huggingface/transfer_learning/"
422+
"llm/prepack/v1.1.1/sourcedir.tar.gz",
421423
"training_prepacked_script_version": "1.1.1",
422424
"training_ecr_specs": {
423425
"framework": "huggingface",
@@ -593,13 +595,15 @@
593595
"g4dn": {
594596
"regional_properties": {"image_uri": "$gpu_ecr_uri_1"},
595597
"properties": {
596-
"gated_model_key_env_var_value": "huggingface-training/g4dn/v1.0.0/train-huggingface-llm-gemma-7b-instruct.tar.gz"
598+
"gated_model_key_env_var_value": "huggingface-training/g4dn/v1.0.0/train-hugg"
599+
"ingface-llm-gemma-7b-instruct.tar.gz"
597600
},
598601
},
599602
"g5": {
600603
"regional_properties": {"image_uri": "$gpu_ecr_uri_1"},
601604
"properties": {
602-
"gated_model_key_env_var_value": "huggingface-training/g5/v1.0.0/train-huggingface-llm-gemma-7b-instruct.tar.gz"
605+
"gated_model_key_env_var_value": "huggingface-training/g5/v1.0.0/train-huggingf"
606+
"ace-llm-gemma-7b-instruct.tar.gz"
603607
},
604608
},
605609
"local_gpu": {"regional_properties": {"image_uri": "$gpu_ecr_uri_1"}},
@@ -608,13 +612,15 @@
608612
"p3dn": {
609613
"regional_properties": {"image_uri": "$gpu_ecr_uri_1"},
610614
"properties": {
611-
"gated_model_key_env_var_value": "huggingface-training/p3dn/v1.0.0/train-huggingface-llm-gemma-7b-instruct.tar.gz"
615+
"gated_model_key_env_var_value": "huggingface-training/p3dn/v1.0.0/train-hugg"
616+
"ingface-llm-gemma-7b-instruct.tar.gz"
612617
},
613618
},
614619
"p4d": {
615620
"regional_properties": {"image_uri": "$gpu_ecr_uri_1"},
616621
"properties": {
617-
"gated_model_key_env_var_value": "huggingface-training/p4d/v1.0.0/train-huggingface-llm-gemma-7b-instruct.tar.gz"
622+
"gated_model_key_env_var_value": "huggingface-training/"
623+
"p4d/v1.0.0/train-huggingface-llm-gemma-7b-instruct.tar.gz"
618624
},
619625
},
620626
"p4de": {"regional_properties": {"image_uri": "$gpu_ecr_uri_1"}},

tests/unit/sagemaker/jumpstart/test_utils.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -906,7 +906,8 @@ def make_accept_eula_inference_spec(*largs, **kwargs):
906906
)
907907
mocked_info_log.assert_any_call(
908908
"Model 'pytorch-eqa-bert-base-cased' requires accepting end-user license agreement (EULA). "
909-
"See https://jumpstart-cache-prod-us-east-1.s3.us-east-1.amazonaws.com/read/the/fine/print.txt for terms of use.",
909+
"See https://jumpstart-cache-prod-us-east-1.s3.us-east-1.amazonaws.com/read/the/fine/print.txt"
910+
" for terms of use.",
910911
)
911912

912913

0 commit comments

Comments
 (0)