From eb5bcdde0a91efeb241d21a0a1ae0dca12c094e6 Mon Sep 17 00:00:00 2001
From: SP Mohanty <spmohanty91@gmail.com>
Date: Tue, 23 Jul 2019 17:58:30 +0200
Subject: [PATCH] Addresses #117 - Refactor env_step in the client
---
flatland/evaluators/client.py | 12 ++++++------
1 file changed, 6 insertions(+), 6 deletions(-)
diff --git a/flatland/evaluators/client.py b/flatland/evaluators/client.py
index e8a8e623..f74af355 100644
--- a/flatland/evaluators/client.py
+++ b/flatland/evaluators/client.py
@@ -187,23 +187,23 @@ class FlatlandRemoteClient(object):
_payload = _response['payload']
# remote_observation = _payload['observation']
- reward = _payload['reward']
- done = _payload['done']
- info = _payload['info']
+ remote_reward = _payload['reward']
+ remote_done = _payload['done']
+ remote_info = _payload['info']
# Replicate the action in the local env
local_observation, local_rewards, local_done, local_info = \
self.env.step(action)
- assert are_dicts_equal(reward, local_rewards)
- assert are_dicts_equal(done, local_done)
+ assert are_dicts_equal(remote_reward, local_rewards)
+ assert are_dicts_equal(remote_done, local_done)
# Return local_observation instead of remote_observation
# as the remote_observation is build using a dummy observation
# builder
# We return the remote rewards and done as they are the
# once used by the evaluator
- return [local_observation, reward, done, info]
+ return [local_observation, remote_reward, remote_done, remote_info]
def submit(self):
_request = {}
--
GitLab