Commit 26267fce authored by Eric Hambro's avatar Eric Hambro
Browse files

Remove unnecessary functions from Batched Agent.

parent 55fb08f9
......@@ -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
raise NotImplementedError
def preprocess_actions(self, actions):
Add any postprocessing steps, for example converting to lists
def batched_step(self):
Return a list of actions
......@@ -32,6 +32,7 @@ def evaluate():
agent = Agent(num_envs, num_actions)
run_batched_rollout(batched_env, agent)
if __name__ == '__main__':
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment