data type changed back

astronomer · Jun 11, 2024 · 3b0dbba · 3b0dbba
1 parent b1fb649
commit 3b0dbba
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/anyscale_provider/operators/anyscale.py b/anyscale_provider/operators/anyscale.py
@@ -151,7 +151,7 @@ def defer_job_polling(self, job_id: str) -> None:
     def get_current_status(self, job_id: str) -> str:
         return str(self.hook.get_job_status(job_id=job_id).state)
 
-    def execute_complete(self, context: Context, event: TriggerEvent) -> None:
+    def execute_complete(self, context: Context, event: Any) -> None:
         current_job_id = event["job_id"]
 
         if event["status"] == JobState.FAILED:
@@ -282,7 +282,7 @@ def execute(self, context: Context) -> Optional[str]:
         self.log.info(f"Service rollout id: {service_id}")        
         return service_id
 
-    def execute_complete(self, context: Context, event: TriggerEvent) -> None:
+    def execute_complete(self, context: Context, event: Any) -> None:
         self.log.info(f"Execution completed...")
         service_id = event["service_name"]