From c52202ad140eff16f95f12ecfb4b46fbe4956b14 Mon Sep 17 00:00:00 2001
From: Tatiana Al-Chueyr <tatiana.alchueyr@gmail.com>
Date: Mon, 2 Dec 2024 12:09:27 +0000
Subject: [PATCH] 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`."
---
 ray_provider/hooks.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ray_provider/hooks.py b/ray_provider/hooks.py
index 2000ecf..38c9132 100644
--- 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"