Offline Datasets res net optionality

lib/objects/map.py

@@ -1,10 +1,9 @@
-import shelve
 from collections import UserDict
 from pathlib import Path
 
 import copy
 from math import sqrt
-from random import choice
+from random import Random
 
 import numpy as np
 
@@ -53,8 +52,12 @@ class Map(object):
             assert array_like_map_representation.ndim == 3
         self.map_array: np.ndarray = array_like_map_representation
         self.name = name
+        self.prng = Random()
         pass
 
+    def seed(self, seed):
+        self.prng.seed(seed)
+
     def __setattr__(self, key, value):
         super(Map, self).__setattr__(key, value)
         if key == 'map_array' and self.map_array is not None:
@@ -102,7 +105,7 @@ class Map(object):
         return trajectory
 
     def get_valid_position(self):
-        valid_position = choice(list(self._G.nodes))
+        valid_position = self.prng.choice(list(self._G.nodes))
         return valid_position
 
     def get_trajectory_from_vertices(self, *args):