Merge pull request #80 from epignatelli/fix-keycorridor

fix: `position_in_room` now creates array properly`
epignatelli · Jul 4, 2024 · 9792102 · 9792102
2 parents 4cf50cc + 6975d11
commit 9792102
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/navix/_version.py b/navix/_version.py
@@ -18,5 +18,5 @@
 # under the License.
 
 
-__version__ = "0.6.13"
+__version__ = "0.6.14"
 __version_info__ = tuple(int(i) for i in __version__.split(".") if i.isdigit())
diff --git a/navix/grid.py b/navix/grid.py
@@ -547,7 +547,7 @@ def position_in_room(self, row: Array, col: Array, *, key: Array) -> Array:
         k1, k2 = jax.random.split(key)
         local_row = jax.random.randint(k1, (), minval=1, maxval=self.room_size[0])
         local_col = jax.random.randint(k2, (), minval=1, maxval=self.room_size[1])
-        return jnp.asarray(local_row, local_col) + self.room_starts[row, col]
+        return jnp.asarray([local_row, local_col]) + self.room_starts[row, col]
 
     @partial(jax.jit, static_argnums=3)
     def position_on_border(