Skip to content

Commit

Permalink
sagemaker.py spell error: tranform change to transform (#34445)
Browse files Browse the repository at this point in the history
* Update README.md library license spell error

* Update README.md spell error,  doctoc  change to  doctor

* Update sagemaker.py  spell error:  tranform change to  transform
  • Loading branch information
yanfangli85 committed Sep 21, 2023
1 parent 659d94f commit 642b4ee
Showing 1 changed file with 7 additions and 7 deletions.
14 changes: 7 additions & 7 deletions airflow/providers/amazon/aws/operators/sagemaker.py
Expand Up @@ -643,7 +643,7 @@ def __init__(
)
self.deferrable = deferrable
self.serialized_model: dict
self.serialized_tranform: dict
self.serialized_transform: dict

def _create_integer_fields(self) -> None:
"""Set fields which should be cast to integers."""
Expand Down Expand Up @@ -716,10 +716,10 @@ def execute(self, context: Context) -> dict:
)

self.serialized_model = serialize(self.hook.describe_model(transform_config["ModelName"]))
self.serialized_tranform = serialize(
self.serialized_transform = serialize(
self.hook.describe_transform_job(transform_config["TransformJobName"])
)
return {"Model": self.serialized_model, "Transform": self.serialized_tranform}
return {"Model": self.serialized_model, "Transform": self.serialized_transform}

def execute_complete(self, context, event=None):
if event["status"] != "success":
Expand All @@ -728,10 +728,10 @@ def execute_complete(self, context, event=None):
self.log.info(event["message"])
transform_config = self.config.get("Transform", self.config)
self.serialized_model = serialize(self.hook.describe_model(transform_config["ModelName"]))
self.serialized_tranform = serialize(
self.serialized_transform = serialize(
self.hook.describe_transform_job(transform_config["TransformJobName"])
)
return {"Model": self.serialized_model, "Transform": self.serialized_tranform}
return {"Model": self.serialized_model, "Transform": self.serialized_transform}

def get_openlineage_facets_on_complete(self, task_instance) -> OperatorLineage:
"""Return OpenLineage data gathered from SageMaker's API response saved by transform job."""
Expand All @@ -747,10 +747,10 @@ def get_openlineage_facets_on_complete(self, task_instance) -> OperatorLineage:
self.log.error("Cannot find Model Package Name.", exc_info=True)

try:
transform_input = self.serialized_tranform["TransformInput"]["DataSource"]["S3DataSource"][
transform_input = self.serialized_transform["TransformInput"]["DataSource"]["S3DataSource"][
"S3Uri"
]
transform_output = self.serialized_tranform["TransformOutput"]["S3OutputPath"]
transform_output = self.serialized_transform["TransformOutput"]["S3OutputPath"]
except KeyError:
self.log.error("Cannot find some required input/output details.", exc_info=True)

Expand Down

0 comments on commit 642b4ee

Please sign in to comment.