Commit a578d2a1 authored by nimishsantosh107's avatar nimishsantosh107
Browse files

readded msgpack-numpy due to test dependencies

parent aa9dd734
Pipeline #8424 failed with stages
in 5 minutes and 35 seconds
......@@ -119,3 +119,5 @@ test_save.dat
.visualizations
playground/
**/tmp
**/TEMP
\ No newline at end of file
......@@ -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)
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