Ninjani commited on
Commit
061f369
·
1 Parent(s): 620a819

include pb individual

Browse files
Files changed (1) hide show
  1. inference_app.py +2 -2
inference_app.py CHANGED
@@ -137,8 +137,8 @@ def get_metrics(
137
  if flexible:
138
  columns.extend(["lddt", "bb_lddt"])
139
  if posebusters:
140
- columns.extend(["posebusters", "posebusters_valid"])
141
-
142
  metrics = metrics[columns].copy()
143
  mapping = {
144
  "lddt_pli_ave": "LDDT-PLI",
 
137
  if flexible:
138
  columns.extend(["lddt", "bb_lddt"])
139
  if posebusters:
140
+ columns.extend([col for col in metrics.columns if col.startswith("posebusters")])
141
+
142
  metrics = metrics[columns].copy()
143
  mapping = {
144
  "lddt_pli_ave": "LDDT-PLI",