Allow units for 'start_pos' (initial agent position)

2022-12-06 12:25:20 +01:00
parent 6d465c69c9
commit 1bf587c4da
2 changed files with 6 additions and 4 deletions
@@ -504,6 +504,8 @@ class ColumbusConfigDefined(ColumbusEnv):
        super().__init__(
            observable=observable, fps=fps, env_seed=env_seed, **kw)
        self.entities_definitions = entities
+        self.start_pos = self.conv_unit(self.start_pos[0], target='em', axis='x'), self.conv_unit(
+            self.start_pos[1], target='em', axis='y')

    def is_unit(self, s):
        if type(s) in [int, float]: