Update code/modbaseModelAdd.py

code/modbaseModelAdd.py

@@ -293,6 +293,6 @@ def addModbaseModels(dataframe, path_to_input_files, path_to_output_files):
     no_modbase_no_Coord = modbase[pd.isna(modbase['coordVAR'])]
     no_modbase = pd.concat([no_modbase, no_modbase_no_Coord])
     modbase = modbase[~pd.isna(modbase['coordVAR'])]
-    st.write(modbase['datapoint', 'model_id', 'sasa'])
+    st.write(modbase[['datapoint', 'model_id', 'sasa']])
     no_modbase = no_modbase[keep_cols]
     return modbase, no_modbase