Merge pull request #99 from astronomer/hotfix

[hotfix] v2.0.2: add try_number to TI
astronomer · Mar 26, 2024 · 849b516 · 849b516
2 parents fbc0c96 + cbf7fea
commit 849b516
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/astronomer_starship/__init__.py b/astronomer_starship/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "2.0.1"
+__version__ = "2.0.2"
 
 
 def get_provider_info():

diff --git a/astronomer_starship/compat/starship_compatability.py b/astronomer_starship/compat/starship_compatability.py
@@ -624,6 +624,7 @@ def task_instances_attrs(cls) -> "Dict[str, AttrDesc]":
                         "run_id": "manual__1970-01-01T00:00:00+00:00",
                         "task_id": "task_id",
                         "map_index": -1,
+                        "try_number": 0,
                         "start_date": epoch_tz,
                         "end_date": epoch_tz,
                         "duration": 1.0,
@@ -678,6 +679,11 @@ def task_instance_attrs(cls) -> "Dict[str, AttrDesc]":
                 "methods": [("POST", True)],
                 "test_value": -1,
             },
+            "try_number": {
+                "attr": "_try_number",
+                "methods": [("POST", True)],
+                "test_value": 0,
+            },
             "start_date": {
                 "attr": "start_date",
                 "methods": [("POST", False)],
@@ -1033,6 +1039,11 @@ class StarshipCompatabilityLayer:
     - 2.2 https://github.com/apache/airflow/tree/2.2.5/airflow/models
     - 2.3 https://github.com/apache/airflow/blob/2.3.4/airflow/models
     - 2.4 https://github.com/apache/airflow/blob/2.4.3/airflow/models
+    - 2.5 https://github.com/apache/airflow/tree/2.5.3/airflow/models
+    - 2.6 https://github.com/apache/airflow/tree/2.6.3/airflow/models
+    - 2.7 https://github.com/apache/airflow/tree/2.7.3/airflow/models
+    - 2.8 https://github.com/apache/airflow/tree/2.8.3/airflow/models
+    - 2.9
 
     >>> isinstance(StarshipCompatabilityLayer("2.8.1"), StarshipAirflow28)
     True