MINOR: Fix athena lineage models (#17662)

open-metadata · Sep 2, 2024 · feacabb · feacabb
1 parent 72afde0
commit feacabb
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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):