plotting

2021-05-25 14:32:37 +02:00
parent 348c4bfecb
commit 3be9ce451d
5 changed files with 84 additions and 57 deletions
--- a/environments/factory/base_factory.py
+++ b/environments/factory/base_factory.py
@ -191,8 +191,7 @@ class BaseFactory(gym.Env):

    def calculate_reward(self, agent_states: List[AgentState]) -> (int, dict):
        # Returns: Reward, Info
-        # Set to "raise NotImplementedError"
-        return 0, {}
+        raise NotImplementedError

    def render(self):
        raise NotImplementedError