Skip to content

Commit

Permalink
Fix type error
Browse files Browse the repository at this point in the history
  • Loading branch information
yifanmai committed Jan 8, 2025
1 parent 4092da4 commit df3c561
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions src/helm/benchmark/scenarios/cti_to_mitre_scenario.py
Original file line number Diff line number Diff line change
Expand Up @@ -159,7 +159,7 @@ def make_label_category_name_dict(jdata: Dict[str, Any]) -> Dict[str, str]:
category_id_to_name[id] = attack["name"]
return category_id_to_name

def get_references(self, num_references: int, correct_cname: str, cnames: List[str]) -> List[str]:
def get_references(self, num_references: int, correct_cname: str, cnames: List[str]) -> List[Reference]:
"""
Randomly select k tec_category_names (attack technique category names) as choices.
However, choose not to include "excluded",
Expand Down Expand Up @@ -196,7 +196,6 @@ def create_multiple_choice_instances(
correct_cname = label_cname[label_tec]
all_cnames = [cname for cname in label_cname.values()]
references = self.get_references(self.num_options, correct_cname, all_cnames)
print(references)
input = Input(text=sentence)
instance = Instance(input, references, split=split)
instances.append(instance)
Expand Down

0 comments on commit df3c561

Please sign in to comment.