Update run_task.py

facebookresearch · Sep 21, 2023 · 624e79d · 624e79d
1 parent c9b25c1
commit 624e79d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/remote_procedure/mnist/run_task.py b/examples/remote_procedure/mnist/run_task.py
@@ -52,7 +52,7 @@ def validate_screening_unit(unit: Unit):
     agent = unit.get_assigned_agent()
     if agent is not None:
         data = agent.state.get_data()
-        annotation = data["final_submission"]["annotations"][0]
+        annotation = data["outputs"]["final_submission"]["annotations"][0]
         if annotation["isCorrect"] and annotation["currentAnnotation"] == 3:
             return True
     return False