fix code

ray-project · Dec 2, 2018 · 2edb020 · 2edb020
1 parent 4760437
commit 2edb020
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/ray/rllib/agents/agent.py b/python/ray/rllib/agents/agent.py
@@ -319,8 +319,8 @@ def train(self):
                 self.local_evaluator.filters))
 
         result = Trainable.train(self)
-        if self.config["callbacks"].get("on_train_step"):
-            self.config["callbacks"]["on_train_step"]({
+        if self.config["callbacks"].get("on_train_result"):
+            self.config["callbacks"]["on_train_result"]({
                 "agent": self,
                 "result": result,
             })