From f561ab6de2fb7f0dad677b42986669d3c476e712 Mon Sep 17 00:00:00 2001
From: Nimish Santosh <nimishsantosh107@icloud.com>
Date: Thu, 12 Aug 2021 04:01:31 +0530
Subject: [PATCH] persistence small fix, enabled timeouts in eval service
---
flatland/envs/persistence.py | 2 +-
flatland/evaluators/service.py | 4 +++-
2 files changed, 4 insertions(+), 2 deletions(-)
diff --git a/flatland/envs/persistence.py b/flatland/envs/persistence.py
index b8354078..41f352e7 100644
--- a/flatland/envs/persistence.py
+++ b/flatland/envs/persistence.py
@@ -163,7 +163,7 @@ class RailEnvPersister(object):
# remove the legacy key
del env_dict["agents_static"]
elif "agents" in env_dict:
- env_dict["agents"] = [EnvAgent(*d[0:12]) for d in env_dict["agents"]]
+ env_dict["agents"] = [EnvAgent(*d[0:len(d)]) for d in env_dict["agents"]]
return env_dict
diff --git a/flatland/evaluators/service.py b/flatland/evaluators/service.py
index 81fa6705..305ee166 100644
--- a/flatland/evaluators/service.py
+++ b/flatland/evaluators/service.py
@@ -10,6 +10,7 @@ import traceback
import json
import itertools
import re
+import logging
import crowdai_api
import msgpack
@@ -130,7 +131,7 @@ class FlatlandRemoteEvaluationService:
shuffle=False,
missing_only=False,
result_output_path=None,
- disable_timeouts=True
+ disable_timeouts=False
):
# Episode recording properties
@@ -725,6 +726,7 @@ class FlatlandRemoteEvaluationService:
del self.env
self.env, _env_dict = RailEnvPersister.load_new(test_env_file_path)
+ # distance map here?
self.begin_simulation = time.time()
--
GitLab