Commit 777f57ce authored by haileyschoelkopf's avatar haileyschoelkopf
Browse files

add comments to all vars defined in evaluate()

parent 9a4e7e7f
...@@ -185,13 +185,22 @@ def evaluate( ...@@ -185,13 +185,22 @@ def evaluate(
# decontaminate = decontamination_ngrams_path is not None # decontaminate = decontamination_ngrams_path is not None
# stores the final result for each task, for each metric/filter pair.
results = collections.defaultdict(dict) results = collections.defaultdict(dict)
# Tracks each task's version.
versions = collections.defaultdict(dict) versions = collections.defaultdict(dict)
# Tracks the YAML configs of all chosen tasks.
configs = collections.defaultdict(dict) configs = collections.defaultdict(dict)
# logs info about each document evaluated.
samples = collections.defaultdict(list) samples = collections.defaultdict(list)
# tracks all Instances/requests a model must generate output on.
requests = collections.defaultdict(list) requests = collections.defaultdict(list)
# ?
aggregate = collections.defaultdict(dict) aggregate = collections.defaultdict(dict)
# tracks if a task was chosen via user selecting a group containing it
task_groups = collections.defaultdict(dict) task_groups = collections.defaultdict(dict)
# stores the amount to pad out reqs per req. type so that
# number of fwd passes per distributed rank is equal
padding_requests = collections.defaultdict(int) padding_requests = collections.defaultdict(int)
# get lists of each type of request # get lists of each type of request
...@@ -200,11 +209,7 @@ def evaluate( ...@@ -200,11 +209,7 @@ def evaluate(
if type(task) == tuple: if type(task) == tuple:
group, task = task group, task = task
# if group in task_groups: task_groups[task_name] = group
# task_groups[group].append(task_name)
# else:
# task_groups[group] = [task_name]
task_groups[task_name] = group
versions[task_name] = task.VERSION versions[task_name] = task.VERSION
configs[task_name] = dict(task.dump_config()) configs[task_name] = dict(task.dump_config())
......
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