Fix RayHook conn_name_attr (#106)

It was mistakely named something else. It overrides the KubernetesHook and all the examples and tests actually use `conn_id` as opposed to `ray_conn_id`."
astronomer · Dec 2, 2024 · c52202a · c52202a
1 parent 8ccc839
commit c52202a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ray_provider/hooks.py b/ray_provider/hooks.py
@@ -26,7 +26,7 @@ class RayHook(KubernetesHook):  # type: ignore
     :param conn_id: The connection ID to use when fetching connection info.
     """
 
-    conn_name_attr = "ray_conn_id"
+    conn_name_attr = "conn_id"
     default_conn_name = "ray_default"
     conn_type = "ray"
     hook_name = "Ray"