visibility

app.py

@@ -38,7 +38,7 @@ def plot_evals_init(model_id, plotly_mode, request: gr.Request):
 def plot_evals(model_id, plotly_mode):
     df = df_cot_err.copy()
     df["selected"] = df_cot_err.model.apply(lambda x: "selected" if x==model_id else "-")
-    df["visible"] = df_cot_err.model.apply(is_visible_model) | df.selected.eq("selected")
+    df["visibility"] = df_cot_err.model.apply(is_visible_model) | df.selected.eq("selected")
     #df.sort_values(["selected", "model"], inplace=True, ascending=True)  # has currently no effect with px.scatter
     template = "plotly_dark" if plotly_mode=="dark" else "plotly" 
     fig = px.scatter(df, x="base accuracy", y="marginal acc. gain", color="selected", symbol="model",
@@ -47,11 +47,12 @@ def plot_evals(model_id, plotly_mode):
                     color_discrete_sequence=["Orange", "Gray"],
                     template=template,
                     error_y="acc_gain-err", hover_data=['model', "cot accuracy"],
+                    custom_data=['model', 'visibility'],
                     width=1200, height=700)
 
     fig.update_traces(
         visible="legendonly",
-        selector=dict(visible=False)
+        selector=dict(visibility=False)
     )
 
     fig.update_layout(