hotfix: diagonal line

AllenNeuralDynamics · Apr 8, 2024 · 9ec86e6 · 9ec86e6
1 parent c9377fd
commit 9ec86e6
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/code/util/streamlit.py b/code/util/streamlit.py
@@ -927,13 +927,14 @@ def _add_agg(df_this, x_name, y_name, group, aggr_method, if_use_x_quantile, q_q
 
     # Add a diagonal line first
     if if_show_diagonal:
-        _min = df[x_name].values.ravel().min()
-        _max = df[y_name].values.ravel().max()
+        _min = np.nanmin(df[x_name].values.ravel())
+        _max = np.nanmax(df[y_name].values.ravel())
         fig.add_trace(go.Scattergl(x=[_min, _max], 
                                    y=[_min, _max], 
                                    mode='lines',
                                    line=dict(dash='dash', color='black', width=2),
-                                   showlegend=False)
+                                   name='x=y',
+                                   showlegend=True)
                       )
 
     for i, group in enumerate(df.sort_values(group_by)[group_by].unique()):