This is an automated email from the ASF dual-hosted git repository.
mobuchowski pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new bbc0968905 openlineage, sagemaker: add missing OpenLineage type
signature (#33114)
bbc0968905 is described below
commit bbc096890512ba2212f318558ca1e954ab399657
Author: Maciej Obuchowski <[email protected]>
AuthorDate: Fri Aug 4 19:11:07 2023 +0200
openlineage, sagemaker: add missing OpenLineage type signature (#33114)
Signed-off-by: Maciej Obuchowski <[email protected]>
---
airflow/providers/amazon/aws/operators/sagemaker.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/airflow/providers/amazon/aws/operators/sagemaker.py
b/airflow/providers/amazon/aws/operators/sagemaker.py
index 173de73933..7e1a0a3567 100644
--- a/airflow/providers/amazon/aws/operators/sagemaker.py
+++ b/airflow/providers/amazon/aws/operators/sagemaker.py
@@ -40,6 +40,8 @@ from airflow.providers.amazon.aws.utils.tags import
format_tags
from airflow.utils.json import AirflowJsonEncoder
if TYPE_CHECKING:
+ from openlineage.client.run import Dataset
+
from airflow.providers.openlineage.extractors.base import OperatorLineage
from airflow.utils.context import Context
@@ -160,7 +162,7 @@ class SageMakerBaseOperator(BaseOperator):
return SageMakerHook(aws_conn_id=self.aws_conn_id)
@staticmethod
- def path_to_s3_dataset(path):
+ def path_to_s3_dataset(path) -> Dataset:
from openlineage.client.run import Dataset
path = path.replace("s3://", "")
