update test

Niccolo-Ajroldi · Niccolo-Ajroldi · commit 107c6b6e2ad3 · 2024-10-03T18:05:01.000+02:00
diff --git a/tests/reference_algorithm_tests.py b/tests/reference_algorithm_tests.py
@@ -471,6 +471,7 @@ def _test_submission(workload_name,
         batch=batch,
         loss_type=workload.loss_type,
         optimizer_state=optimizer_state,
+        train_state={},
         eval_results=[],
         global_step=global_step,
         rng=update_rng)
