Quick adjustments.

HARPLab · Jun 16, 2022 · 88a288e · 88a288e
1 parent af6ab26
commit 88a288e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/tests/corl22.py b/tests/corl22.py
@@ -25,7 +25,7 @@
     eval_start_time = time.strftime("_%m:%d:%H:%M", time.localtime())
     for agent, name in zip(agents, agent_names):
         print("Evaluating " + name + " agent...                    ")
-        perf, dist, q_type, dempref_metric = Evaluation.run(
+        perf, dist, q_type = Evaluation.run(
             domain,
             teacher,
             agent,

diff --git a/tests/evaluation.py b/tests/evaluation.py
@@ -32,6 +32,7 @@ def run(
         init_w_rand = RandomState(0)
         real_num_queries = num_queries
         if actual_queries != None:
+            print(f"Asked for {real_num_queries} but stopping after {actual_queries}.")
             num_queries = actual_queries
         perf_mat = np.zeros(
             (num_tasks, num_runs, num_test_states, num_queries + 1)