diff --git a/ingestion/src/metadata/ingestion/source/database/athena/models.py b/ingestion/src/metadata/ingestion/source/database/athena/models.py index b03a8eca91b4..3a2ea622ad96 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):