Added return type to task rule

astronomer · Sep 5, 2024 · 5e7e1ca · 5e7e1ca
1 parent 91ca09e
commit 5e7e1ca
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/orbiter_translations/automic/xml_demo.py b/orbiter_translations/automic/xml_demo.py
@@ -302,7 +302,9 @@ def bash_command_rule(
 
 
 @task_rule(priority=2)
-def start_or_end_rule(val):
+def start_or_end_rule(
+    val: dict,
+) -> OrbiterEmptyOperator | None:
     """
     Translate input into an EmptyOperator for start and end tasks