Skip to content

Commit

Permalink
[Minor] fix styles
Browse files Browse the repository at this point in the history
  • Loading branch information
huyiwen committed May 27, 2024
1 parent 3d66f5b commit 72f8677
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
2 changes: 1 addition & 1 deletion utilization/dataset/xnli.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ class Xnli(MultipleChoiceDataset):
example_set = "train"
load_args = ("xnli",)
banned_subsets = ["all_languages"]

def init_arguments(self):
from langcodes import Language
self.language = Language(self.subset_name).language_name("en")
Expand Down
3 changes: 2 additions & 1 deletion utilization/dataset/xwinograd.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@

logger = getLogger(__name__)


class Xwinograd(MultipleChoiceDataset):
"""The dataset of XWinograd.
Expand All @@ -25,7 +26,7 @@ class Xwinograd(MultipleChoiceDataset):
instruction = "Given the sentence '{{sentence.strip()}}' in {{lang}}, fill in the blank with the appropriate option: who does '_' refer to?{{'\n'+options if options}}\nAnswer:"
evaluation_set = "test"
load_args = ("Muennighoff/xwinograd",)

def init_arguments(self):
from langcodes import Language
self.language = Language(self.subset_name).language_name("en")
Expand Down

0 comments on commit 72f8677

Please sign in to comment.