Skip to content

GitLab

Commit 1bc6cdb1 authored by Jonathan Tow's avatar Jonathan Tow
Browse files

Add type info to doc-string

parent 744482bb
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
lm_eval/evaluator.py
View file @ 1bc6cdb1
...@@ -32,7 +32,7 @@ def simple_evaluate(model, model_args, task_names, ...@@ -32,7 +32,7 @@ def simple_evaluate(model, model_args, task_names,
Limit the number of examples per task (only use this for testing) Limit the number of examples per task (only use this for testing)
:param bootstrap_iters: :param bootstrap_iters:
Number of iterations for bootstrap statistics Number of iterations for bootstrap statistics
:param description_dict: :param description_dict: dict[str, str]
Dictionary of custom task descriptions of the form: `task_name: description` Dictionary of custom task descriptions of the form: `task_name: description`
:return :return
Dictionary of results Dictionary of results
...@@ -83,7 +83,7 @@ def evaluate(lm, task_dict, provide_description, num_fewshot, limit, bootstrap_i ...@@ -83,7 +83,7 @@ def evaluate(lm, task_dict, provide_description, num_fewshot, limit, bootstrap_i
Limit the number of examples per task (only use this for testing) Limit the number of examples per task (only use this for testing)
:param bootstrap_iters: :param bootstrap_iters:
Number of iterations for bootstrap statistics Number of iterations for bootstrap statistics
:param description_dict: :param description_dict: dict[str, str]
Dictionary of custom task descriptions of the form: `task_name: description` Dictionary of custom task descriptions of the form: `task_name: description`
:return :return
Dictionary of results Dictionary of results
......
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