@@ -1117,8 +1117,8 @@ def test_get_codeartifact_command(pipeline_session):
1117
1117
)
1118
1118
1119
1119
codeartifact_command = processor ._get_codeartifact_command (codeartifact_repo_arn = codeartifact_repo_arn )
1120
-
1121
- assert codeartifact_command == "aws codeartifact login --tool pip --domain test-domain --domain-owner 012345678901 --repository test-repository --region us-west-2"
1120
+
1121
+ assert codeartifact_command == "aws codeartifact login --tool pip --domain test-domain --domain-owner 012345678901 --repository test-repository --region us-west-2" # noqa: E501
1122
1122
1123
1123
1124
1124
@patch ("sagemaker.workflow.utilities._pipeline_config" , MOCKED_PIPELINE_CONFIG )
@@ -1137,6 +1137,7 @@ def test_get_codeartifact_command_bad_repo_arn(pipeline_session):
1137
1137
with pytest .raises (ValueError ):
1138
1138
processor ._get_codeartifact_command (codeartifact_repo_arn = codeartifact_repo_arn )
1139
1139
1140
+
1140
1141
@patch ("sagemaker.workflow.utilities._pipeline_config" , MOCKED_PIPELINE_CONFIG )
1141
1142
def test_generate_framework_script (pipeline_session ):
1142
1143
processor = PyTorchProcessor (
@@ -1177,7 +1178,8 @@ def test_generate_framework_script(pipeline_session):
1177
1178
python process.py "$@"
1178
1179
"""
1179
1180
)
1180
-
1181
+
1182
+
1181
1183
@patch ("sagemaker.workflow.utilities._pipeline_config" , MOCKED_PIPELINE_CONFIG )
1182
1184
def test_generate_framework_script_with_codeartifact (pipeline_session ):
1183
1185
processor = PyTorchProcessor (
@@ -1209,7 +1211,7 @@ def test_generate_framework_script_with_codeartifact(pipeline_session):
1209
1211
if ! hash aws 2>/dev/null; then
1210
1212
echo "AWS CLI is not installed. Skipping CodeArtifact login."
1211
1213
else
1212
- " aws codeartifact login --tool pip --domain test-domain --domain-owner 012345678901 --repository test-repository --region us-west-2"
1214
+ aws codeartifact login --tool pip --domain test-domain --domain-owner 012345678901 --repository test-repository --region us-west-2
1213
1215
fi
1214
1216
1215
1217
# Some py3 containers has typing, which may breaks pip install
@@ -1219,9 +1221,10 @@ def test_generate_framework_script_with_codeartifact(pipeline_session):
1219
1221
fi
1220
1222
1221
1223
python process.py "$@"
1222
- """
1224
+ """ # noqa: E501
1223
1225
)
1224
1226
1227
+
1225
1228
def _get_script_processor (sagemaker_session ):
1226
1229
return ScriptProcessor (
1227
1230
role = ROLE ,
0 commit comments