MINOR: Fix athena lineage models (#17662)

This commit is contained in:
Mayur Singal 2024-09-02 14:19:13 +05:30 committed by ulixius9
parent 72afde0aca
commit feacabb2f3

View File

@ -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):