From a578d2a16a61bdb05598c8174ae6b63dbdfc9329 Mon Sep 17 00:00:00 2001
From: nimishsantosh107 <nimishsantosh107@icloud.com>
Date: Wed, 8 Sep 2021 11:08:52 +0530
Subject: [PATCH] readded msgpack-numpy due to test dependencies
---
.gitignore | 2 ++
flatland/envs/persistence.py | 4 +++-
2 files changed, 5 insertions(+), 1 deletion(-)
diff --git a/.gitignore b/.gitignore
index 2f1f81d1..ce15e015 100644
--- a/.gitignore
+++ b/.gitignore
@@ -119,3 +119,5 @@ test_save.dat
.visualizations
playground/
+**/tmp
+**/TEMP
\ No newline at end of file
diff --git a/flatland/envs/persistence.py b/flatland/envs/persistence.py
index ca23b8d1..188ac7c2 100644
--- a/flatland/envs/persistence.py
+++ b/flatland/envs/persistence.py
@@ -2,6 +2,7 @@
import pickle
import msgpack
+import msgpack_numpy
import numpy as np
from flatland.envs import rail_env
@@ -22,6 +23,7 @@ from flatland.envs import malfunction_generators as mal_gen
from flatland.envs import rail_generators as rail_gen
from flatland.envs import line_generators as line_gen
+msgpack_numpy.patch()
class RailEnvPersister(object):
@@ -308,4 +310,4 @@ class RailEnvPersister(object):
self.height, self.width = self.rail.grid.shape
self.rail.height = self.height
self.rail.width = self.width
- self.dones = dict.fromkeys(list(range(self.get_num_agents())) + ["__all__"], False)
\ No newline at end of file
+ self.dones = dict.fromkeys(list(range(self.get_num_agents())) + ["__all__"], False)
--
GitLab