diff --git a/providers/src/airflow/providers/amazon/aws/operators/athena.py b/providers/src/airflow/providers/amazon/aws/operators/athena.py index d48ac751d4239..c602611803ed5 100644 --- a/providers/src/airflow/providers/amazon/aws/operators/athena.py +++ b/providers/src/airflow/providers/amazon/aws/operators/athena.py @@ -311,7 +311,9 @@ def get_openlineage_dataset(self, database, table) -> Dataset | None: } fields = [ SchemaDatasetFacetFields( - name=column["Name"], type=column["Type"], description=column["Comment"] + name=column["Name"], + type=column["Type"], + description=column.get("Comment"), ) for column in table_metadata["TableMetadata"]["Columns"] ] diff --git a/providers/tests/amazon/aws/operators/athena_metadata.json b/providers/tests/amazon/aws/operators/athena_metadata.json index f13b1241742ab..121a51920c9f0 100644 --- a/providers/tests/amazon/aws/operators/athena_metadata.json +++ b/providers/tests/amazon/aws/operators/athena_metadata.json @@ -28,8 +28,7 @@ }, { "Name": "ENDS_ON", - "Type": "timestamp", - "Comment": "from deserializer" + "Type": "timestamp" } ], "PartitionKeys": [], diff --git a/providers/tests/amazon/aws/operators/test_athena.py b/providers/tests/amazon/aws/operators/test_athena.py index e4d7d6622a091..5ee557caea906 100644 --- a/providers/tests/amazon/aws/operators/test_athena.py +++ b/providers/tests/amazon/aws/operators/test_athena.py @@ -358,7 +358,7 @@ def mock_get_table_metadata(CatalogName, DatabaseName, TableName): SchemaDatasetFacetFields( name="ENDS_ON", type="timestamp", - description="from deserializer", + description=None, ), ], ),