Update model_comparison.py

model_comparison.py

@@ -77,21 +77,23 @@ def dataframe_with_selections(df):
 def add_user_evalID_columns_to_df(df, evalDataPath):
     # with open(evalDataPath, 'r') as f:
     # yamlData = safe_load(f)
-    yamlData = safe_load(hf_hub_download(repo_id="JVice/try-before-you-bias-data", filename=evalDataPath, repo_type="dataset"))
-    st.write(yamlData)
-    for user in yamlData['evaluations']['username']:
-        if df is None:
-            df = pd.DataFrame(yamlData['evaluations']['username'][user]).T
-            df.insert(0, "Eval. ID", list(yamlData['evaluations']['username'][user].keys()), True)
-        else:
-            df = pd.concat([df, pd.DataFrame(yamlData['evaluations']['username'][user]).T],
-                                   ignore_index=True)
-        evalIDIterator = 0
-        for index, row in df.iterrows():
-            if row['Eval. ID'] is np.nan:
-                df.loc[index, 'Eval. ID'] = list(yamlData['evaluations']['username'][user].keys())[
-                    evalIDIterator]
-                evalIDIterator += 1
+    # yamlData = safe_load()
+    with open(hf_hub_download(repo_id="JVice/try-before-you-bias-data", filename=evalDataPath, repo_type="dataset"), 'r') as f:
+        yamlData = safe_load(f)
+        st.write(yamlData)
+        for user in yamlData['evaluations']['username']:
+            if df is None:
+                df = pd.DataFrame(yamlData['evaluations']['username'][user]).T
+                df.insert(0, "Eval. ID", list(yamlData['evaluations']['username'][user].keys()), True)
+            else:
+                df = pd.concat([df, pd.DataFrame(yamlData['evaluations']['username'][user]).T],
+                                       ignore_index=True)
+            evalIDIterator = 0
+            for index, row in df.iterrows():
+                if row['Eval. ID'] is np.nan:
+                    df.loc[index, 'Eval. ID'] = list(yamlData['evaluations']['username'][user].keys())[
+                        evalIDIterator]
+                    evalIDIterator += 1
     return df
     
 def initialise_page(tab):