Tuan Tran commited on
Commit
25e429e
·
1 Parent(s): 4f9529a

add a patch for atack variant name

Browse files
Files changed (1) hide show
  1. backend/examples.py +6 -2
backend/examples.py CHANGED
@@ -167,8 +167,12 @@ def build_infos(abs_path: Path, datatype: str, dataset_name: str, db_key: str):
167
  if attack_variant == "default":
168
  attack = attack_name
169
  else:
170
- attack = f"{attack_name}_{attack_variant}"
171
-
 
 
 
 
172
  if len(attack_rows) == 0:
173
  model_infos[attack] = []
174
  continue
 
167
  if attack_variant == "default":
168
  attack = attack_name
169
  else:
170
+ # TODO: Update data on S3 with new Omniseal Bench V2 eval script
171
+ if str(abs_path).startswith("http") or str(abs_path).startswith("https") or str(abs_path).startswith("s3://"):
172
+ attack = f"{attack_name}_{attack_variant}"
173
+ else:
174
+ attack = f"{attack_name}__{attack_variant}"
175
+
176
  if len(attack_rows) == 0:
177
  model_infos[attack] = []
178
  continue