fixed tooltip error

utils/viz.py

@@ -205,8 +205,14 @@ def create_scatter_plot(df, x: str, y: str, x_label: str = None, y_label: str =
         agent_data = df[df[hover_data[0]] == agent]
         
         # remove url from tooltip name
-        agent_data.loc[:, 'Agent Name'] = agent_data['Agent Name'].str.rsplit(']').str[0].str[1:] if '[' in agent_data['Agent Name'] else agent_data['Agent Name']
+        def clean_agent_name(name):
+            if '[' in str(name):
+                return name[1:].rsplit(']')[0]
+            return name
 
+        # Apply the function to each element individually
+        agent_data.loc[:, 'Agent Name'] = agent_data['Agent Name'].apply(clean_agent_name)
+        
         x_value = [np.mean(agent_data[x].values)]
         y_value = [np.mean(agent_data[y].values)]