Remove unnecessary functions from Batched Agent.

agents/batched_agent.py

@@ -11,20 +11,9 @@ class BatchedAgent:
         self.num_envs = num_envs
         self.num_actions = num_actions
 
-    def preprocess_observations(self, observations, rewards, dones, infos):
+    def batched_step(self, observations, rewards, dones, infos):
         """
-        Add any preprocessing steps, for example rerodering/stacking for torch/tf in your model
+        Take list of outputs of each environments and return a list of actions
         """
-        pass
+        raise NotImplementedError
 
-    def preprocess_actions(self, actions):
-        """
-        Add any postprocessing steps, for example converting to lists
-        """
-        pass
-
-    def batched_step(self):
-        """
-        Return a list of actions
-        """
-        pass

local_evaluation.py

@@ -32,6 +32,7 @@ def evaluate():
     agent = Agent(num_envs, num_actions)
 
     run_batched_rollout(batched_env, agent)
+    
 
 if __name__ == '__main__':
     evaluate()