Version upgrade and a couple of temporary changes

astronomer · Nov 6, 2024 · 020fabf · 020fabf
1 parent fd44077
commit 020fabf
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/astronomer_starship/__init__.py b/astronomer_starship/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "2.0.6"
+__version__ = "2.1.0"
 
 
 def get_provider_info():

diff --git a/astronomer_starship/compat/starship_compatability.py b/astronomer_starship/compat/starship_compatability.py
@@ -1037,7 +1037,7 @@ def task_instance_attrs(self):
         attrs["custom_operator_name"] = {
             "attr": "custom_operator_name",
             "methods": [("POST", True)],
-            "test_value": "@task",
+            "test_value": None,
         }
         return attrs
 
@@ -1080,6 +1080,7 @@ def task_instance_attrs(self):
             "methods": [("POST", False)],
             "test_value": "task_display_name",
         }
+        return attrs
 
 
 class StarshipAirflow210(StarshipAirflow28):

diff --git a/tests/validation_test.py b/tests/validation_test.py
@@ -27,6 +27,8 @@
 ]
 
 IMAGES = [
+    "apache/airflow:slim-2.10.3",
+    "apache/airflow:slim-2.9.3",
     "apache/airflow:slim-2.8.1",
     "apache/airflow:slim-2.7.3",
     "apache/airflow:slim-2.6.0",
@@ -75,7 +77,7 @@ def run_in_container(
     volumes: list,
     environment: dict,
     platform: str,
-) -> (int, bytes):
+) -> tuple[int, bytes]:
     from docker.models.containers import Container
 
     container: Container = docker_client.containers.run(