Skip to content

Commit fbca6da

Browse files
fix: typo and remove local test changes
1 parent 3bc4298 commit fbca6da

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

src/sagemaker/workflow/pipeline.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,7 @@ def describe(self) -> Dict[str, Any]:
143143
"""
144144
return self.sagemaker_session.sagemaker_client.describe_pipeline(PipelineName=self.name)
145145

146-
def update(self, role_arn: str, description: str = None, ) -> Dict[str, Any]:
146+
def update(self, role_arn: str, description: str = None) -> Dict[str, Any]:
147147
"""Updates a Pipeline in the Workflow service.
148148
149149
Args:

tests/integ/test_workflow.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -410,13 +410,14 @@ def test_three_step_definition(
410410

411411
def test_one_step_sklearn_processing_pipeline(
412412
sagemaker_session,
413+
role,
413414
sklearn_latest_version,
414415
cpu_instance_type,
415416
pipeline_name,
416417
region_name,
417418
athena_dataset_definition,
418419
):
419-
role = "arn:aws:iam::734680132978:role/SageMakerFullAccessRole"
420+
role
420421
instance_count = ParameterInteger(name="InstanceCount", default_value=2)
421422
script_path = os.path.join(DATA_DIR, "dummy_script.py")
422423
input_file_path = os.path.join(DATA_DIR, "dummy_input.txt")

0 commit comments

Comments
 (0)