"include/ck/ck.hpp" did not exist on "81c942cd7e198b46f55be31451d51f27459d34ab"
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(
# decontaminate = decontamination_ngrams_path is not None
# stores the final result for each task, for each metric/filter pair.
results = collections.defaultdict(dict)
# Tracks each task's version.
versions = collections.defaultdict(dict)
# Tracks the YAML configs of all chosen tasks.
configs = collections.defaultdict(dict)
# logs info about each document evaluated.
samples = collections.defaultdict(list)
# tracks all Instances/requests a model must generate output on.
requests = collections.defaultdict(list)
# ?
aggregate = collections.defaultdict(dict)
# tracks if a task was chosen via user selecting a group containing it
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)
# get lists of each type of request
......@@ -200,11 +209,7 @@ def evaluate(
if type(task) == tuple:
group, task = task
# if group in task_groups:
# task_groups[group].append(task_name)
# else:
# task_groups[group] = [task_name]
task_groups[task_name] = group
task_groups[task_name] = group
versions[task_name] = task.VERSION
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