From d15f8aecd61067756b7bd83d4171fb67f5208759 Mon Sep 17 00:00:00 2001 From: Wei Lee Date: Tue, 9 Apr 2024 16:12:31 +0800 Subject: [PATCH] style(core): fix mypy warning --- astronomer/providers/core/sensors/filesystem.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/astronomer/providers/core/sensors/filesystem.py b/astronomer/providers/core/sensors/filesystem.py index 1ed91f13f..53258d5a5 100644 --- a/astronomer/providers/core/sensors/filesystem.py +++ b/astronomer/providers/core/sensors/filesystem.py @@ -1,7 +1,9 @@ +from __future__ import annotations + import os import warnings from datetime import timedelta -from typing import Any, Dict, Optional +from typing import Any from airflow.hooks.filesystem import FSHook from airflow.sensors.filesystem import FileSensor @@ -54,7 +56,7 @@ def execute(self, context: Context) -> None: method_name="execute_complete", ) - def execute_complete(self, context: Context, event: Optional[Dict[str, Any]]) -> None: + def execute_complete(self, context: Context, event: bool | None = None) -> None: """ Callback for when the trigger fires - returns immediately. Relies on trigger to throw an exception, otherwise it assumes execution was