From 9ec86e653079f8abb6b4fe3fa78aaf857283b8c0 Mon Sep 17 00:00:00 2001 From: hh_itx_win10 Date: Mon, 8 Apr 2024 01:47:20 -0700 Subject: [PATCH] hotfix: diagonal line --- code/util/streamlit.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/code/util/streamlit.py b/code/util/streamlit.py index 17c68f0..73dc971 100644 --- 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()):