From feacabb2f3362bde4533e0f00a56a53ddd6e3802 Mon Sep 17 00:00:00 2001 From: Mayur Singal <39544459+ulixius9@users.noreply.github.com> Date: Mon, 2 Sep 2024 14:19:13 +0530 Subject: [PATCH] MINOR: Fix athena lineage models (#17662) --- .../metadata/ingestion/source/database/athena/models.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/ingestion/src/metadata/ingestion/source/database/athena/models.py b/ingestion/src/metadata/ingestion/source/database/athena/models.py index b03a8eca91b..3a2ea622ad9 100644 --- a/ingestion/src/metadata/ingestion/source/database/athena/models.py +++ b/ingestion/src/metadata/ingestion/source/database/athena/models.py @@ -22,19 +22,19 @@ class QueryExecutionIdsResponse(BaseModel): QueryExecutionIds: Optional[List[str]] = None -class Status(BaseModel): +class AthenaStatus(BaseModel): State: Optional[str] = "FAILED" # Default value SubmissionDateTime: Optional[datetime] = None -class Statistics(BaseModel): +class AthenaStatistics(BaseModel): TotalExecutionTimeInMillis: Optional[int] = None class AthenaQueryExecution(BaseModel): Query: Optional[str] = None - Statistics: Optional[Statistics] = None - Status: Optional[Status] = None + Statistics: Optional[AthenaStatistics] = None + Status: Optional[AthenaStatus] = None class AthenaQueryExecutionList(BaseModel):