adjust interface to reflect actual signature (#374)

Co-authored-by: Jeremy Reynolds <jeremr@microsoft.com>
microsoft · Jul 15, 2021 · 69e2127 · 69e2127
1 parent a61968b
commit 69e2127
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/maro/rl/training/trajectory.py b/maro/rl/training/trajectory.py
@@ -18,7 +18,7 @@ def get_action(self, action_by_agent, event) -> dict:
     def get_reward(self) -> float:
         pass
 
-    def on_env_feedback(self):
+    def on_env_feedback(self, event, state_by_agent, action_by_agent, reward):
         pass
 
     def on_finish(self):