Skip to content
Snippets Groups Projects
Commit a578d2a1 authored by nimishsantosh107's avatar nimishsantosh107
Browse files

readded msgpack-numpy due to test dependencies

parent aa9dd734
No related branches found
No related tags found
No related merge requests found
...@@ -119,3 +119,5 @@ test_save.dat ...@@ -119,3 +119,5 @@ test_save.dat
.visualizations .visualizations
playground/ playground/
**/tmp
**/TEMP
\ No newline at end of file
...@@ -2,6 +2,7 @@ ...@@ -2,6 +2,7 @@
import pickle import pickle
import msgpack import msgpack
import msgpack_numpy
import numpy as np import numpy as np
from flatland.envs import rail_env from flatland.envs import rail_env
...@@ -22,6 +23,7 @@ from flatland.envs import malfunction_generators as mal_gen ...@@ -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 rail_generators as rail_gen
from flatland.envs import line_generators as line_gen from flatland.envs import line_generators as line_gen
msgpack_numpy.patch()
class RailEnvPersister(object): class RailEnvPersister(object):
...@@ -308,4 +310,4 @@ class RailEnvPersister(object): ...@@ -308,4 +310,4 @@ class RailEnvPersister(object):
self.height, self.width = self.rail.grid.shape self.height, self.width = self.rail.grid.shape
self.rail.height = self.height self.rail.height = self.height
self.rail.width = self.width self.rail.width = self.width
self.dones = dict.fromkeys(list(range(self.get_num_agents())) + ["__all__"], False) self.dones = dict.fromkeys(list(range(self.get_num_agents())) + ["__all__"], False)
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment