Fixed plot

app.py

@@ -3,6 +3,7 @@ import pandas as pd
 import os
 import fnmatch
 import json
+import plotly.express as px
 
 class MultiURLData:
     def __init__(self):
@@ -74,23 +75,21 @@ selected_models = st.multiselect(
 filtered_data = data_provider.get_data(selected_models)
 st.dataframe(filtered_data)
 
-
-
-#TODO fix this plot.  currently has an error
 # Create a plot with new data
 df = pd.DataFrame({
     'Model': list(filtered_data['Model Name']),
-    'harness|arc:challenge|25_rank': list(filtered_data['harness|arc:challenge|25_rank']),
-    'harness|moral_scenarios|5_rank': list(filtered_data['harness|moral_scenarios|5_rank']),
+    # use debug to troubheshoot error
+    'harness|arc:challenge|25': list(filtered_data['harness|arc:challenge|25']),
+    'harness|moral_scenarios|5': list(filtered_data['harness|moral_scenarios|5']),
 })
 
 # Calculate color column
 df['color'] = 'purple'
-df.loc[df['harness|moral_scenarios|5_rank'] < df['harness|arc:challenge|25_rank'], 'color'] = 'red'
-df.loc[df['harness|moral_scenarios|5_rank'] > df['harness|arc:challenge|25_rank'], 'color'] = 'blue'
+df.loc[df['harness|moral_scenarios|5'] < df['harness|arc:challenge|25'], 'color'] = 'red'
+df.loc[df['harness|moral_scenarios|5'] > df['harness|arc:challenge|25'], 'color'] = 'blue'
 
 # Create the scatter plot
-fig = px.scatter(df, x='harness|arc:challenge|25_rank', y='harness|moral_scenarios|5_rank', color='color', hover_data=['Model'])
+fig = px.scatter(df, x='harness|arc:challenge|25', y='harness|moral_scenarios|5', color='color', hover_data=['Model'])
 fig.update_layout(showlegend=False,  # hide legend
                   xaxis = dict(autorange="reversed"),  # reverse X-axis
                   yaxis = dict(autorange="reversed"))  # reverse Y-axis