Commit d762e2aa authored by Baber's avatar Baber
Browse files

fix

parent 7a8203fa
from rich.traceback import install
from lm_eval._cli.harness import HarnessCLI from lm_eval._cli.harness import HarnessCLI
from lm_eval.utils import setup_logging from lm_eval.utils import setup_logging
install(show_locals=True)
def cli_evaluate() -> None: def cli_evaluate() -> None:
"""Main CLI entry point.""" """Main CLI entry point."""
setup_logging() setup_logging()
......
...@@ -660,6 +660,8 @@ class ConfigurableTask(Task): ...@@ -660,6 +660,8 @@ class ConfigurableTask(Task):
if hasattr(self, _method): if hasattr(self, _method):
setattr(self, _method, MethodType(fn, self)) setattr(self, _method, MethodType(fn, self))
self.runtime_checks(self.task_docs[0])
print("hello") print("hello")
def download(self, dataset_kwargs: dict[str, Any] | None = None, **kwargs) -> None: def download(self, dataset_kwargs: dict[str, Any] | None = None, **kwargs) -> None:
...@@ -1427,6 +1429,7 @@ class ConfigurableTask(Task): ...@@ -1427,6 +1429,7 @@ class ConfigurableTask(Task):
# Test One Doc # Test One Doc
self.features: list[str] = list(self.task_docs.features.keys()) self.features: list[str] = list(self.task_docs.features.keys())
self.multiple_target = 0 self.multiple_target = 0
self.multiple_input = 0
test_text = self.doc_to_text(test_doc) test_text = self.doc_to_text(test_doc)
test_target = self.doc_to_target(test_doc) test_target = self.doc_to_target(test_doc)
...@@ -1434,13 +1437,19 @@ class ConfigurableTask(Task): ...@@ -1434,13 +1437,19 @@ class ConfigurableTask(Task):
test_choice = self.doc_to_choice(test_doc) test_choice = self.doc_to_choice(test_doc)
if not isinstance(test_choice, list): if not isinstance(test_choice, list):
eval_logger.error("doc_to_choice must return list") eval_logger.error("doc_to_choice must return list")
# else: else:
# num_choice = len(test_choice) num_choice = len(test_choice)
if isinstance(test_text, int):
eval_logger.debug(
"doc_to_text returned an int. Assuming multiple inputs."
)
if isinstance(test_text, int): if isinstance(test_text, int):
eval_logger.debug( eval_logger.debug(
"doc_to_text returned an int. Assuming multiple inputs." "doc_to_text returned an int. Assuming multiple inputs."
) )
self.multiple_input = num_choice
else: else:
test_choice = None test_choice = None
......
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