Skip to content

GitLab

Commit e9e5305d authored by Shashi456's avatar Shashi456
Browse files

Turning on data flags

parent f42f168c
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 7 deletions
lm_eval/tasks/mlsum.py
View file @ e9e5305d
...@@ -28,13 +28,13 @@ class MLSUMEs(PromptSourceTask): ...@@ -28,13 +28,13 @@ class MLSUMEs(PromptSourceTask):
DATASET_NAME = "es" DATASET_NAME = "es"
def has_training_docs(self): def has_training_docs(self):
return False return True
def has_validation_docs(self): def has_validation_docs(self):
return False return True
def has_test_docs(self): def has_test_docs(self):
return False return True
def training_docs(self): def training_docs(self):
if self.has_training_docs(): if self.has_training_docs():
...@@ -64,13 +64,13 @@ class MLSUMDe(PromptSourceTask): ...@@ -64,13 +64,13 @@ class MLSUMDe(PromptSourceTask):
DATASET_NAME = "de" DATASET_NAME = "de"
def has_training_docs(self): def has_training_docs(self):
return False return True
def has_validation_docs(self): def has_validation_docs(self):
return False return True
def has_test_docs(self): def has_test_docs(self):
return False return True
def training_docs(self): def training_docs(self):
if self.has_training_docs(): if self.has_training_docs():
...@@ -90,4 +90,4 @@ class MLSUMDe(PromptSourceTask): ...@@ -90,4 +90,4 @@ class MLSUMDe(PromptSourceTask):
return "." return "."
def max_generation_length(self): def max_generation_length(self):
return 120 return 120
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment