mettagrid 0.1.0__py3-none-any.whl → 0.1.2__py3-none-any.whl

build/mettagrid/mettagrid.cpp

@@ -3860,6 +3860,7 @@ static const char __pyx_k_wall[] = "wall";
 static const char __pyx_k_ASCII[] = "ASCII";
 static const char __pyx_k_agent[] = "agent.";
 static const char __pyx_k_altar[] = "altar";
+static const char __pyx_k_block[] = "block";
 static const char __pyx_k_class[] = "__class__";
 static const char __pyx_k_count[] = "count";
 static const char __pyx_k_dtype[] = "dtype";
@@ -3915,6 +3916,7 @@ static const char __pyx_k_Sequence[] = "Sequence";
 static const char __pyx_k_agent_id[] = "agent_id";
 static const char __pyx_k_getstate[] = "__getstate__";
 static const char __pyx_k_itemsize[] = "itemsize";
+static const char __pyx_k_mine_red[] = "mine.red";
 static const char __pyx_k_pyx_type[] = "__pyx_type";
 static const char __pyx_k_register[] = "register";
 static const char __pyx_k_setstate[] = "__setstate__";
@@ -3957,7 +3959,9 @@ static const char __pyx_k_to_container[] = "to_container";
 static const char __pyx_k_version_info[] = "version_info";
 static const char __pyx_k_AttackNearest[] = "AttackNearest";
 static const char __pyx_k_class_getitem[] = "__class_getitem__";
+static const char __pyx_k_generator_red[] = "generator.red";
 static const char __pyx_k_grid_features[] = "grid_features";
+static const char __pyx_k_objects_block[] = "objects.block";
 static const char __pyx_k_reduce_cython[] = "__reduce_cython__";
 static const char __pyx_k_AssertionError[] = "AssertionError";
 static const char __pyx_k_MettaGrid_step[] = "MettaGrid.step";
@@ -4288,6 +4292,8 @@ typedef struct {
   PyObject *__pyx_n_s_asyncio_coroutines;
   PyObject *__pyx_n_s_attack;
   PyObject *__pyx_n_s_base;
+  PyObject *__pyx_n_s_block;
+  PyObject *__pyx_n_u_block;
   PyObject *__pyx_n_s_c;
   PyObject *__pyx_n_u_c;
   PyObject *__pyx_n_s_change_color;
@@ -4320,8 +4326,8 @@ typedef struct {
   PyObject *__pyx_n_s_game;
   PyObject *__pyx_n_u_game;
   PyObject *__pyx_kp_u_gc;
-  PyObject *__pyx_n_s_generator;
   PyObject *__pyx_n_u_generator;
+  PyObject *__pyx_kp_u_generator_red;
   PyObject *__pyx_n_s_get;
   PyObject *__pyx_n_s_get_episode_stats;
   PyObject *__pyx_n_s_get_items;
@@ -4369,8 +4375,8 @@ typedef struct {
   PyObject *__pyx_n_s_mettagrid_actions_swap;
   PyObject *__pyx_n_s_mettagrid_mettagrid_c;
   PyObject *__pyx_kp_s_mettagrid_mettagrid_pyx;
-  PyObject *__pyx_n_s_mine;
   PyObject *__pyx_n_u_mine;
+  PyObject *__pyx_kp_u_mine_red;
   PyObject *__pyx_n_s_mode;
   PyObject *__pyx_n_s_move;
   PyObject *__pyx_n_s_name;
@@ -4632,6 +4638,8 @@ static int __pyx_m_clear(PyObject *m) {
   Py_CLEAR(clear_module_state->__pyx_n_s_asyncio_coroutines);
   Py_CLEAR(clear_module_state->__pyx_n_s_attack);
   Py_CLEAR(clear_module_state->__pyx_n_s_base);
+  Py_CLEAR(clear_module_state->__pyx_n_s_block);
+  Py_CLEAR(clear_module_state->__pyx_n_u_block);
   Py_CLEAR(clear_module_state->__pyx_n_s_c);
   Py_CLEAR(clear_module_state->__pyx_n_u_c);
   Py_CLEAR(clear_module_state->__pyx_n_s_change_color);
@@ -4664,8 +4672,8 @@ static int __pyx_m_clear(PyObject *m) {
   Py_CLEAR(clear_module_state->__pyx_n_s_game);
   Py_CLEAR(clear_module_state->__pyx_n_u_game);
   Py_CLEAR(clear_module_state->__pyx_kp_u_gc);
-  Py_CLEAR(clear_module_state->__pyx_n_s_generator);
   Py_CLEAR(clear_module_state->__pyx_n_u_generator);
+  Py_CLEAR(clear_module_state->__pyx_kp_u_generator_red);
   Py_CLEAR(clear_module_state->__pyx_n_s_get);
   Py_CLEAR(clear_module_state->__pyx_n_s_get_episode_stats);
   Py_CLEAR(clear_module_state->__pyx_n_s_get_items);
@@ -4713,8 +4721,8 @@ static int __pyx_m_clear(PyObject *m) {
   Py_CLEAR(clear_module_state->__pyx_n_s_mettagrid_actions_swap);
   Py_CLEAR(clear_module_state->__pyx_n_s_mettagrid_mettagrid_c);
   Py_CLEAR(clear_module_state->__pyx_kp_s_mettagrid_mettagrid_pyx);
-  Py_CLEAR(clear_module_state->__pyx_n_s_mine);
   Py_CLEAR(clear_module_state->__pyx_n_u_mine);
+  Py_CLEAR(clear_module_state->__pyx_kp_u_mine_red);
   Py_CLEAR(clear_module_state->__pyx_n_s_mode);
   Py_CLEAR(clear_module_state->__pyx_n_s_move);
   Py_CLEAR(clear_module_state->__pyx_n_s_name);
@@ -4954,6 +4962,8 @@ static int __pyx_m_traverse(PyObject *m, visitproc visit, void *arg) {
   Py_VISIT(traverse_module_state->__pyx_n_s_asyncio_coroutines);
   Py_VISIT(traverse_module_state->__pyx_n_s_attack);
   Py_VISIT(traverse_module_state->__pyx_n_s_base);
+  Py_VISIT(traverse_module_state->__pyx_n_s_block);
+  Py_VISIT(traverse_module_state->__pyx_n_u_block);
   Py_VISIT(traverse_module_state->__pyx_n_s_c);
   Py_VISIT(traverse_module_state->__pyx_n_u_c);
   Py_VISIT(traverse_module_state->__pyx_n_s_change_color);
@@ -4986,8 +4996,8 @@ static int __pyx_m_traverse(PyObject *m, visitproc visit, void *arg) {
   Py_VISIT(traverse_module_state->__pyx_n_s_game);
   Py_VISIT(traverse_module_state->__pyx_n_u_game);
   Py_VISIT(traverse_module_state->__pyx_kp_u_gc);
-  Py_VISIT(traverse_module_state->__pyx_n_s_generator);
   Py_VISIT(traverse_module_state->__pyx_n_u_generator);
+  Py_VISIT(traverse_module_state->__pyx_kp_u_generator_red);
   Py_VISIT(traverse_module_state->__pyx_n_s_get);
   Py_VISIT(traverse_module_state->__pyx_n_s_get_episode_stats);
   Py_VISIT(traverse_module_state->__pyx_n_s_get_items);
@@ -5035,8 +5045,8 @@ static int __pyx_m_traverse(PyObject *m, visitproc visit, void *arg) {
   Py_VISIT(traverse_module_state->__pyx_n_s_mettagrid_actions_swap);
   Py_VISIT(traverse_module_state->__pyx_n_s_mettagrid_mettagrid_c);
   Py_VISIT(traverse_module_state->__pyx_kp_s_mettagrid_mettagrid_pyx);
-  Py_VISIT(traverse_module_state->__pyx_n_s_mine);
   Py_VISIT(traverse_module_state->__pyx_n_u_mine);
+  Py_VISIT(traverse_module_state->__pyx_kp_u_mine_red);
   Py_VISIT(traverse_module_state->__pyx_n_s_mode);
   Py_VISIT(traverse_module_state->__pyx_n_s_move);
   Py_VISIT(traverse_module_state->__pyx_n_s_name);
@@ -5358,6 +5368,8 @@ static int __pyx_m_traverse(PyObject *m, visitproc visit, void *arg) {
 #define __pyx_n_s_asyncio_coroutines __pyx_mstate_global->__pyx_n_s_asyncio_coroutines
 #define __pyx_n_s_attack __pyx_mstate_global->__pyx_n_s_attack
 #define __pyx_n_s_base __pyx_mstate_global->__pyx_n_s_base
+#define __pyx_n_s_block __pyx_mstate_global->__pyx_n_s_block
+#define __pyx_n_u_block __pyx_mstate_global->__pyx_n_u_block
 #define __pyx_n_s_c __pyx_mstate_global->__pyx_n_s_c
 #define __pyx_n_u_c __pyx_mstate_global->__pyx_n_u_c
 #define __pyx_n_s_change_color __pyx_mstate_global->__pyx_n_s_change_color
@@ -5390,8 +5402,8 @@ static int __pyx_m_traverse(PyObject *m, visitproc visit, void *arg) {
 #define __pyx_n_s_game __pyx_mstate_global->__pyx_n_s_game
 #define __pyx_n_u_game __pyx_mstate_global->__pyx_n_u_game
 #define __pyx_kp_u_gc __pyx_mstate_global->__pyx_kp_u_gc
-#define __pyx_n_s_generator __pyx_mstate_global->__pyx_n_s_generator
 #define __pyx_n_u_generator __pyx_mstate_global->__pyx_n_u_generator
+#define __pyx_kp_u_generator_red __pyx_mstate_global->__pyx_kp_u_generator_red
 #define __pyx_n_s_get __pyx_mstate_global->__pyx_n_s_get
 #define __pyx_n_s_get_episode_stats __pyx_mstate_global->__pyx_n_s_get_episode_stats
 #define __pyx_n_s_get_items __pyx_mstate_global->__pyx_n_s_get_items
@@ -5439,8 +5451,8 @@ static int __pyx_m_traverse(PyObject *m, visitproc visit, void *arg) {
 #define __pyx_n_s_mettagrid_actions_swap __pyx_mstate_global->__pyx_n_s_mettagrid_actions_swap
 #define __pyx_n_s_mettagrid_mettagrid_c __pyx_mstate_global->__pyx_n_s_mettagrid_mettagrid_c
 #define __pyx_kp_s_mettagrid_mettagrid_pyx __pyx_mstate_global->__pyx_kp_s_mettagrid_mettagrid_pyx
-#define __pyx_n_s_mine __pyx_mstate_global->__pyx_n_s_mine
 #define __pyx_n_u_mine __pyx_mstate_global->__pyx_n_u_mine
+#define __pyx_kp_u_mine_red __pyx_mstate_global->__pyx_kp_u_mine_red
 #define __pyx_n_s_mode __pyx_mstate_global->__pyx_n_s_mode
 #define __pyx_n_s_move __pyx_mstate_global->__pyx_n_s_move
 #define __pyx_n_s_name __pyx_mstate_global->__pyx_n_s_name
@@ -22016,6 +22028,8 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
   PyObject *__pyx_v_r = NULL;
   PyObject *__pyx_v_c = NULL;
   Wall *__pyx_v_wall;
+  Wall *__pyx_v_block;
+  PyObject *__pyx_v_m = NULL;
   PyObject *__pyx_v_agent_cfg = NULL;
   PyObject *__pyx_v_rewards = NULL;
   std::string __pyx_v_inv_item;
@@ -23308,7 +23322,7 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
  *                     wall = new Wall(r, c, cfg.objects.wall)
  *                     self._grid.add_object(wall)             # <<<<<<<<<<<<<<
  *                     self._stats.incr(b"objects.wall")
- * 
+ *                 elif map[r,c] == "block":
  */
         (void)(__pyx_v_self->__pyx_base._grid->add_object(__pyx_v_wall));
 
@@ -23316,8 +23330,8 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
  *                     wall = new Wall(r, c, cfg.objects.wall)
  *                     self._grid.add_object(wall)
  *                     self._stats.incr(b"objects.wall")             # <<<<<<<<<<<<<<
- * 
- *                 elif map[r,c] == "mine":
+ *                 elif map[r,c] == "block":
+ *                     block = new Wall(r, c, cfg.objects.block)
  */
         __pyx_v_self->__pyx_base._stats.incr(__pyx_k_objects_wall);
 
@@ -23331,161 +23345,385 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
         goto __pyx_L28;
       }
 
-      /* "mettagrid/mettagrid.pyx":118
+      /* "mettagrid/mettagrid.pyx":117
+ *                     self._grid.add_object(wall)
  *                     self._stats.incr(b"objects.wall")
- * 
- *                 elif map[r,c] == "mine":             # <<<<<<<<<<<<<<
- *                     converter = new Mine(r, c, cfg.objects.mine)
- *                 elif map[r,c] == "generator":
+ *                 elif map[r,c] == "block":             # <<<<<<<<<<<<<<
+ *                     block = new Wall(r, c, cfg.objects.block)
+ *                     self._grid.add_object(block)
  */
-      __pyx_t_10 = PyTuple_New(2); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 118, __pyx_L1_error)
+      __pyx_t_10 = PyTuple_New(2); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 117, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_10);
       __Pyx_INCREF(__pyx_v_r);
       __Pyx_GIVEREF(__pyx_v_r);
-      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 0, __pyx_v_r)) __PYX_ERR(0, 118, __pyx_L1_error);
+      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 0, __pyx_v_r)) __PYX_ERR(0, 117, __pyx_L1_error);
       __Pyx_INCREF(__pyx_v_c);
       __Pyx_GIVEREF(__pyx_v_c);
-      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 1, __pyx_v_c)) __PYX_ERR(0, 118, __pyx_L1_error);
-      __pyx_t_11 = __Pyx_PyObject_GetItem(__pyx_v_map, __pyx_t_10); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 118, __pyx_L1_error)
+      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 1, __pyx_v_c)) __PYX_ERR(0, 117, __pyx_L1_error);
+      __pyx_t_11 = __Pyx_PyObject_GetItem(__pyx_v_map, __pyx_t_10); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 117, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_11);
       __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
-      __pyx_t_6 = (__Pyx_PyUnicode_Equals(__pyx_t_11, __pyx_n_u_mine, Py_EQ)); if (unlikely((__pyx_t_6 < 0))) __PYX_ERR(0, 118, __pyx_L1_error)
+      __pyx_t_6 = (__Pyx_PyUnicode_Equals(__pyx_t_11, __pyx_n_u_block, Py_EQ)); if (unlikely((__pyx_t_6 < 0))) __PYX_ERR(0, 117, __pyx_L1_error)
       __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
       if (__pyx_t_6) {
 
+        /* "mettagrid/mettagrid.pyx":118
+ *                     self._stats.incr(b"objects.wall")
+ *                 elif map[r,c] == "block":
+ *                     block = new Wall(r, c, cfg.objects.block)             # <<<<<<<<<<<<<<
+ *                     self._grid.add_object(block)
+ *                     self._stats.incr(b"objects.block")
+ */
+        __pyx_t_23 = __Pyx_PyInt_As_GridCoord(__pyx_v_r); if (unlikely((__pyx_t_23 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 118, __pyx_L1_error)
+        __pyx_t_22 = __Pyx_PyInt_As_GridCoord(__pyx_v_c); if (unlikely((__pyx_t_22 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 118, __pyx_L1_error)
+        __pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_v_cfg, __pyx_n_s_objects); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 118, __pyx_L1_error)
+        __Pyx_GOTREF(__pyx_t_11);
+        __pyx_t_10 = __Pyx_PyObject_GetAttrStr(__pyx_t_11, __pyx_n_s_block); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 118, __pyx_L1_error)
+        __Pyx_GOTREF(__pyx_t_10);
+        __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
+        __pyx_t_24 = __pyx_convert_map_from_py_std_3a__3a_string__and_int(__pyx_t_10); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 118, __pyx_L1_error)
+        __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
+        try {
+          __pyx_t_25 = new Wall(__pyx_t_23, __pyx_t_22, __pyx_t_24);
+        } catch(...) {
+          __Pyx_CppExn2PyErr();
+          __PYX_ERR(0, 118, __pyx_L1_error)
+        }
+        __pyx_v_block = __pyx_t_25;
+
         /* "mettagrid/mettagrid.pyx":119
- * 
- *                 elif map[r,c] == "mine":
- *                     converter = new Mine(r, c, cfg.objects.mine)             # <<<<<<<<<<<<<<
- *                 elif map[r,c] == "generator":
- *                     converter = new Generator(r, c, cfg.objects.generator)
+ *                 elif map[r,c] == "block":
+ *                     block = new Wall(r, c, cfg.objects.block)
+ *                     self._grid.add_object(block)             # <<<<<<<<<<<<<<
+ *                     self._stats.incr(b"objects.block")
+ *                 elif map[r,c].startswith("mine"):
+ */
+        (void)(__pyx_v_self->__pyx_base._grid->add_object(__pyx_v_block));
+
+        /* "mettagrid/mettagrid.pyx":120
+ *                     block = new Wall(r, c, cfg.objects.block)
+ *                     self._grid.add_object(block)
+ *                     self._stats.incr(b"objects.block")             # <<<<<<<<<<<<<<
+ *                 elif map[r,c].startswith("mine"):
+ *                     m = map[r,c]
+ */
+        __pyx_v_self->__pyx_base._stats.incr(__pyx_k_objects_block);
+
+        /* "mettagrid/mettagrid.pyx":117
+ *                     self._grid.add_object(wall)
+ *                     self._stats.incr(b"objects.wall")
+ *                 elif map[r,c] == "block":             # <<<<<<<<<<<<<<
+ *                     block = new Wall(r, c, cfg.objects.block)
+ *                     self._grid.add_object(block)
  */
-        __pyx_t_23 = __Pyx_PyInt_As_GridCoord(__pyx_v_r); if (unlikely((__pyx_t_23 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 119, __pyx_L1_error)
-        __pyx_t_22 = __Pyx_PyInt_As_GridCoord(__pyx_v_c); if (unlikely((__pyx_t_22 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 119, __pyx_L1_error)
-        __pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_v_cfg, __pyx_n_s_objects); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 119, __pyx_L1_error)
+        goto __pyx_L28;
+      }
+
+      /* "mettagrid/mettagrid.pyx":121
+ *                     self._grid.add_object(block)
+ *                     self._stats.incr(b"objects.block")
+ *                 elif map[r,c].startswith("mine"):             # <<<<<<<<<<<<<<
+ *                     m = map[r,c]
+ *                     if "." not in m:
+ */
+      __pyx_t_11 = PyTuple_New(2); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 121, __pyx_L1_error)
+      __Pyx_GOTREF(__pyx_t_11);
+      __Pyx_INCREF(__pyx_v_r);
+      __Pyx_GIVEREF(__pyx_v_r);
+      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_11, 0, __pyx_v_r)) __PYX_ERR(0, 121, __pyx_L1_error);
+      __Pyx_INCREF(__pyx_v_c);
+      __Pyx_GIVEREF(__pyx_v_c);
+      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_11, 1, __pyx_v_c)) __PYX_ERR(0, 121, __pyx_L1_error);
+      __pyx_t_1 = __Pyx_PyObject_GetItem(__pyx_v_map, __pyx_t_11); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 121, __pyx_L1_error)
+      __Pyx_GOTREF(__pyx_t_1);
+      __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
+      __pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_startswith); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 121, __pyx_L1_error)
+      __Pyx_GOTREF(__pyx_t_11);
+      __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
+      __pyx_t_1 = NULL;
+      __pyx_t_5 = 0;
+      #if CYTHON_UNPACK_METHODS
+      if (likely(PyMethod_Check(__pyx_t_11))) {
+        __pyx_t_1 = PyMethod_GET_SELF(__pyx_t_11);
+        if (likely(__pyx_t_1)) {
+          PyObject* function = PyMethod_GET_FUNCTION(__pyx_t_11);
+          __Pyx_INCREF(__pyx_t_1);
+          __Pyx_INCREF(function);
+          __Pyx_DECREF_SET(__pyx_t_11, function);
+          __pyx_t_5 = 1;
+        }
+      }
+      #endif
+      {
+        PyObject *__pyx_callargs[2] = {__pyx_t_1, __pyx_n_u_mine};
+        __pyx_t_10 = __Pyx_PyObject_FastCall(__pyx_t_11, __pyx_callargs+1-__pyx_t_5, 1+__pyx_t_5);
+        __Pyx_XDECREF(__pyx_t_1); __pyx_t_1 = 0;
+        if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 121, __pyx_L1_error)
+        __Pyx_GOTREF(__pyx_t_10);
+        __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
+      }
+      __pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_10); if (unlikely((__pyx_t_6 < 0))) __PYX_ERR(0, 121, __pyx_L1_error)
+      __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
+      if (__pyx_t_6) {
+
+        /* "mettagrid/mettagrid.pyx":122
+ *                     self._stats.incr(b"objects.block")
+ *                 elif map[r,c].startswith("mine"):
+ *                     m = map[r,c]             # <<<<<<<<<<<<<<
+ *                     if "." not in m:
+ *                         m = "mine.red"
+ */
+        __pyx_t_10 = PyTuple_New(2); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 122, __pyx_L1_error)
+        __Pyx_GOTREF(__pyx_t_10);
+        __Pyx_INCREF(__pyx_v_r);
+        __Pyx_GIVEREF(__pyx_v_r);
+        if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 0, __pyx_v_r)) __PYX_ERR(0, 122, __pyx_L1_error);
+        __Pyx_INCREF(__pyx_v_c);
+        __Pyx_GIVEREF(__pyx_v_c);
+        if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 1, __pyx_v_c)) __PYX_ERR(0, 122, __pyx_L1_error);
+        __pyx_t_11 = __Pyx_PyObject_GetItem(__pyx_v_map, __pyx_t_10); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 122, __pyx_L1_error)
+        __Pyx_GOTREF(__pyx_t_11);
+        __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
+        __Pyx_XDECREF_SET(__pyx_v_m, __pyx_t_11);
+        __pyx_t_11 = 0;
+
+        /* "mettagrid/mettagrid.pyx":123
+ *                 elif map[r,c].startswith("mine"):
+ *                     m = map[r,c]
+ *                     if "." not in m:             # <<<<<<<<<<<<<<
+ *                         m = "mine.red"
+ *                     converter = new Mine(r, c, cfg.objects[m])
+ */
+        __pyx_t_6 = (__Pyx_PySequence_ContainsTF(__pyx_kp_u__2, __pyx_v_m, Py_NE)); if (unlikely((__pyx_t_6 < 0))) __PYX_ERR(0, 123, __pyx_L1_error)
+        if (__pyx_t_6) {
+
+          /* "mettagrid/mettagrid.pyx":124
+ *                     m = map[r,c]
+ *                     if "." not in m:
+ *                         m = "mine.red"             # <<<<<<<<<<<<<<
+ *                     converter = new Mine(r, c, cfg.objects[m])
+ *                 elif map[r,c].startswith("generator"):
+ */
+          __Pyx_INCREF(__pyx_kp_u_mine_red);
+          __Pyx_DECREF_SET(__pyx_v_m, __pyx_kp_u_mine_red);
+
+          /* "mettagrid/mettagrid.pyx":123
+ *                 elif map[r,c].startswith("mine"):
+ *                     m = map[r,c]
+ *                     if "." not in m:             # <<<<<<<<<<<<<<
+ *                         m = "mine.red"
+ *                     converter = new Mine(r, c, cfg.objects[m])
+ */
+        }
+
+        /* "mettagrid/mettagrid.pyx":125
+ *                     if "." not in m:
+ *                         m = "mine.red"
+ *                     converter = new Mine(r, c, cfg.objects[m])             # <<<<<<<<<<<<<<
+ *                 elif map[r,c].startswith("generator"):
+ *                     m = map[r,c]
+ */
+        __pyx_t_22 = __Pyx_PyInt_As_GridCoord(__pyx_v_r); if (unlikely((__pyx_t_22 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 125, __pyx_L1_error)
+        __pyx_t_23 = __Pyx_PyInt_As_GridCoord(__pyx_v_c); if (unlikely((__pyx_t_23 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 125, __pyx_L1_error)
+        __pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_v_cfg, __pyx_n_s_objects); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 125, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_11);
-        __pyx_t_10 = __Pyx_PyObject_GetAttrStr(__pyx_t_11, __pyx_n_s_mine); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 119, __pyx_L1_error)
+        __pyx_t_10 = __Pyx_PyObject_GetItem(__pyx_t_11, __pyx_v_m); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 125, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_10);
         __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
-        __pyx_t_24 = __pyx_convert_map_from_py_std_3a__3a_string__and_int(__pyx_t_10); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 119, __pyx_L1_error)
+        __pyx_t_24 = __pyx_convert_map_from_py_std_3a__3a_string__and_int(__pyx_t_10); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 125, __pyx_L1_error)
         __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
         try {
-          __pyx_t_26 = new Mine(__pyx_t_23, __pyx_t_22, __pyx_t_24);
+          __pyx_t_26 = new Mine(__pyx_t_22, __pyx_t_23, __pyx_t_24);
         } catch(...) {
           __Pyx_CppExn2PyErr();
-          __PYX_ERR(0, 119, __pyx_L1_error)
+          __PYX_ERR(0, 125, __pyx_L1_error)
         }
         __pyx_v_converter = __pyx_t_26;
 
-        /* "mettagrid/mettagrid.pyx":118
- *                     self._stats.incr(b"objects.wall")
- * 
- *                 elif map[r,c] == "mine":             # <<<<<<<<<<<<<<
- *                     converter = new Mine(r, c, cfg.objects.mine)
- *                 elif map[r,c] == "generator":
+        /* "mettagrid/mettagrid.pyx":121
+ *                     self._grid.add_object(block)
+ *                     self._stats.incr(b"objects.block")
+ *                 elif map[r,c].startswith("mine"):             # <<<<<<<<<<<<<<
+ *                     m = map[r,c]
+ *                     if "." not in m:
  */
         goto __pyx_L28;
       }
 
-      /* "mettagrid/mettagrid.pyx":120
- *                 elif map[r,c] == "mine":
- *                     converter = new Mine(r, c, cfg.objects.mine)
- *                 elif map[r,c] == "generator":             # <<<<<<<<<<<<<<
- *                     converter = new Generator(r, c, cfg.objects.generator)
- *                 elif map[r,c] == "altar":
+      /* "mettagrid/mettagrid.pyx":126
+ *                         m = "mine.red"
+ *                     converter = new Mine(r, c, cfg.objects[m])
+ *                 elif map[r,c].startswith("generator"):             # <<<<<<<<<<<<<<
+ *                     m = map[r,c]
+ *                     if "." not in m:
  */
-      __pyx_t_10 = PyTuple_New(2); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 120, __pyx_L1_error)
-      __Pyx_GOTREF(__pyx_t_10);
+      __pyx_t_11 = PyTuple_New(2); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 126, __pyx_L1_error)
+      __Pyx_GOTREF(__pyx_t_11);
       __Pyx_INCREF(__pyx_v_r);
       __Pyx_GIVEREF(__pyx_v_r);
-      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 0, __pyx_v_r)) __PYX_ERR(0, 120, __pyx_L1_error);
+      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_11, 0, __pyx_v_r)) __PYX_ERR(0, 126, __pyx_L1_error);
       __Pyx_INCREF(__pyx_v_c);
       __Pyx_GIVEREF(__pyx_v_c);
-      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 1, __pyx_v_c)) __PYX_ERR(0, 120, __pyx_L1_error);
-      __pyx_t_11 = __Pyx_PyObject_GetItem(__pyx_v_map, __pyx_t_10); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 120, __pyx_L1_error)
+      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_11, 1, __pyx_v_c)) __PYX_ERR(0, 126, __pyx_L1_error);
+      __pyx_t_1 = __Pyx_PyObject_GetItem(__pyx_v_map, __pyx_t_11); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 126, __pyx_L1_error)
+      __Pyx_GOTREF(__pyx_t_1);
+      __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
+      __pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_startswith); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 126, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_11);
+      __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
+      __pyx_t_1 = NULL;
+      __pyx_t_5 = 0;
+      #if CYTHON_UNPACK_METHODS
+      if (likely(PyMethod_Check(__pyx_t_11))) {
+        __pyx_t_1 = PyMethod_GET_SELF(__pyx_t_11);
+        if (likely(__pyx_t_1)) {
+          PyObject* function = PyMethod_GET_FUNCTION(__pyx_t_11);
+          __Pyx_INCREF(__pyx_t_1);
+          __Pyx_INCREF(function);
+          __Pyx_DECREF_SET(__pyx_t_11, function);
+          __pyx_t_5 = 1;
+        }
+      }
+      #endif
+      {
+        PyObject *__pyx_callargs[2] = {__pyx_t_1, __pyx_n_u_generator};
+        __pyx_t_10 = __Pyx_PyObject_FastCall(__pyx_t_11, __pyx_callargs+1-__pyx_t_5, 1+__pyx_t_5);
+        __Pyx_XDECREF(__pyx_t_1); __pyx_t_1 = 0;
+        if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 126, __pyx_L1_error)
+        __Pyx_GOTREF(__pyx_t_10);
+        __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
+      }
+      __pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_10); if (unlikely((__pyx_t_6 < 0))) __PYX_ERR(0, 126, __pyx_L1_error)
       __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
-      __pyx_t_6 = (__Pyx_PyUnicode_Equals(__pyx_t_11, __pyx_n_u_generator, Py_EQ)); if (unlikely((__pyx_t_6 < 0))) __PYX_ERR(0, 120, __pyx_L1_error)
-      __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
       if (__pyx_t_6) {
 
-        /* "mettagrid/mettagrid.pyx":121
- *                     converter = new Mine(r, c, cfg.objects.mine)
- *                 elif map[r,c] == "generator":
- *                     converter = new Generator(r, c, cfg.objects.generator)             # <<<<<<<<<<<<<<
+        /* "mettagrid/mettagrid.pyx":127
+ *                     converter = new Mine(r, c, cfg.objects[m])
+ *                 elif map[r,c].startswith("generator"):
+ *                     m = map[r,c]             # <<<<<<<<<<<<<<
+ *                     if "." not in m:
+ *                         m = "generator.red"
+ */
+        __pyx_t_10 = PyTuple_New(2); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 127, __pyx_L1_error)
+        __Pyx_GOTREF(__pyx_t_10);
+        __Pyx_INCREF(__pyx_v_r);
+        __Pyx_GIVEREF(__pyx_v_r);
+        if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 0, __pyx_v_r)) __PYX_ERR(0, 127, __pyx_L1_error);
+        __Pyx_INCREF(__pyx_v_c);
+        __Pyx_GIVEREF(__pyx_v_c);
+        if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 1, __pyx_v_c)) __PYX_ERR(0, 127, __pyx_L1_error);
+        __pyx_t_11 = __Pyx_PyObject_GetItem(__pyx_v_map, __pyx_t_10); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 127, __pyx_L1_error)
+        __Pyx_GOTREF(__pyx_t_11);
+        __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
+        __Pyx_XDECREF_SET(__pyx_v_m, __pyx_t_11);
+        __pyx_t_11 = 0;
+
+        /* "mettagrid/mettagrid.pyx":128
+ *                 elif map[r,c].startswith("generator"):
+ *                     m = map[r,c]
+ *                     if "." not in m:             # <<<<<<<<<<<<<<
+ *                         m = "generator.red"
+ *                     converter = new Generator(r, c, cfg.objects[m])
+ */
+        __pyx_t_6 = (__Pyx_PySequence_ContainsTF(__pyx_kp_u__2, __pyx_v_m, Py_NE)); if (unlikely((__pyx_t_6 < 0))) __PYX_ERR(0, 128, __pyx_L1_error)
+        if (__pyx_t_6) {
+
+          /* "mettagrid/mettagrid.pyx":129
+ *                     m = map[r,c]
+ *                     if "." not in m:
+ *                         m = "generator.red"             # <<<<<<<<<<<<<<
+ *                     converter = new Generator(r, c, cfg.objects[m])
+ *                 elif map[r,c] == "altar":
+ */
+          __Pyx_INCREF(__pyx_kp_u_generator_red);
+          __Pyx_DECREF_SET(__pyx_v_m, __pyx_kp_u_generator_red);
+
+          /* "mettagrid/mettagrid.pyx":128
+ *                 elif map[r,c].startswith("generator"):
+ *                     m = map[r,c]
+ *                     if "." not in m:             # <<<<<<<<<<<<<<
+ *                         m = "generator.red"
+ *                     converter = new Generator(r, c, cfg.objects[m])
+ */
+        }
+
+        /* "mettagrid/mettagrid.pyx":130
+ *                     if "." not in m:
+ *                         m = "generator.red"
+ *                     converter = new Generator(r, c, cfg.objects[m])             # <<<<<<<<<<<<<<
  *                 elif map[r,c] == "altar":
  *                     converter = new Altar(r, c, cfg.objects.altar)
  */
-        __pyx_t_22 = __Pyx_PyInt_As_GridCoord(__pyx_v_r); if (unlikely((__pyx_t_22 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 121, __pyx_L1_error)
-        __pyx_t_23 = __Pyx_PyInt_As_GridCoord(__pyx_v_c); if (unlikely((__pyx_t_23 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 121, __pyx_L1_error)
-        __pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_v_cfg, __pyx_n_s_objects); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 121, __pyx_L1_error)
+        __pyx_t_23 = __Pyx_PyInt_As_GridCoord(__pyx_v_r); if (unlikely((__pyx_t_23 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 130, __pyx_L1_error)
+        __pyx_t_22 = __Pyx_PyInt_As_GridCoord(__pyx_v_c); if (unlikely((__pyx_t_22 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 130, __pyx_L1_error)
+        __pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_v_cfg, __pyx_n_s_objects); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 130, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_11);
-        __pyx_t_10 = __Pyx_PyObject_GetAttrStr(__pyx_t_11, __pyx_n_s_generator); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 121, __pyx_L1_error)
+        __pyx_t_10 = __Pyx_PyObject_GetItem(__pyx_t_11, __pyx_v_m); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 130, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_10);
         __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
-        __pyx_t_24 = __pyx_convert_map_from_py_std_3a__3a_string__and_int(__pyx_t_10); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 121, __pyx_L1_error)
+        __pyx_t_24 = __pyx_convert_map_from_py_std_3a__3a_string__and_int(__pyx_t_10); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 130, __pyx_L1_error)
         __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
-        __pyx_v_converter = new Generator(__pyx_t_22, __pyx_t_23, __pyx_t_24);
+        __pyx_v_converter = new Generator(__pyx_t_23, __pyx_t_22, __pyx_t_24);
 
-        /* "mettagrid/mettagrid.pyx":120
- *                 elif map[r,c] == "mine":
- *                     converter = new Mine(r, c, cfg.objects.mine)
- *                 elif map[r,c] == "generator":             # <<<<<<<<<<<<<<
- *                     converter = new Generator(r, c, cfg.objects.generator)
- *                 elif map[r,c] == "altar":
+        /* "mettagrid/mettagrid.pyx":126
+ *                         m = "mine.red"
+ *                     converter = new Mine(r, c, cfg.objects[m])
+ *                 elif map[r,c].startswith("generator"):             # <<<<<<<<<<<<<<
+ *                     m = map[r,c]
+ *                     if "." not in m:
  */
         goto __pyx_L28;
       }
 
-      /* "mettagrid/mettagrid.pyx":122
- *                 elif map[r,c] == "generator":
- *                     converter = new Generator(r, c, cfg.objects.generator)
+      /* "mettagrid/mettagrid.pyx":131
+ *                         m = "generator.red"
+ *                     converter = new Generator(r, c, cfg.objects[m])
  *                 elif map[r,c] == "altar":             # <<<<<<<<<<<<<<
  *                     converter = new Altar(r, c, cfg.objects.altar)
  *                 elif map[r,c] == "armory":
  */
-      __pyx_t_10 = PyTuple_New(2); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 122, __pyx_L1_error)
+      __pyx_t_10 = PyTuple_New(2); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 131, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_10);
       __Pyx_INCREF(__pyx_v_r);
       __Pyx_GIVEREF(__pyx_v_r);
-      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 0, __pyx_v_r)) __PYX_ERR(0, 122, __pyx_L1_error);
+      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 0, __pyx_v_r)) __PYX_ERR(0, 131, __pyx_L1_error);
       __Pyx_INCREF(__pyx_v_c);
       __Pyx_GIVEREF(__pyx_v_c);
-      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 1, __pyx_v_c)) __PYX_ERR(0, 122, __pyx_L1_error);
-      __pyx_t_11 = __Pyx_PyObject_GetItem(__pyx_v_map, __pyx_t_10); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 122, __pyx_L1_error)
+      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 1, __pyx_v_c)) __PYX_ERR(0, 131, __pyx_L1_error);
+      __pyx_t_11 = __Pyx_PyObject_GetItem(__pyx_v_map, __pyx_t_10); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 131, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_11);
       __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
-      __pyx_t_6 = (__Pyx_PyUnicode_Equals(__pyx_t_11, __pyx_n_u_altar, Py_EQ)); if (unlikely((__pyx_t_6 < 0))) __PYX_ERR(0, 122, __pyx_L1_error)
+      __pyx_t_6 = (__Pyx_PyUnicode_Equals(__pyx_t_11, __pyx_n_u_altar, Py_EQ)); if (unlikely((__pyx_t_6 < 0))) __PYX_ERR(0, 131, __pyx_L1_error)
       __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
       if (__pyx_t_6) {
 
-        /* "mettagrid/mettagrid.pyx":123
- *                     converter = new Generator(r, c, cfg.objects.generator)
+        /* "mettagrid/mettagrid.pyx":132
+ *                     converter = new Generator(r, c, cfg.objects[m])
  *                 elif map[r,c] == "altar":
  *                     converter = new Altar(r, c, cfg.objects.altar)             # <<<<<<<<<<<<<<
  *                 elif map[r,c] == "armory":
  *                     converter = new Armory(r, c, cfg.objects.armory)
  */
-        __pyx_t_23 = __Pyx_PyInt_As_GridCoord(__pyx_v_r); if (unlikely((__pyx_t_23 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 123, __pyx_L1_error)
-        __pyx_t_22 = __Pyx_PyInt_As_GridCoord(__pyx_v_c); if (unlikely((__pyx_t_22 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 123, __pyx_L1_error)
-        __pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_v_cfg, __pyx_n_s_objects); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 123, __pyx_L1_error)
+        __pyx_t_22 = __Pyx_PyInt_As_GridCoord(__pyx_v_r); if (unlikely((__pyx_t_22 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 132, __pyx_L1_error)
+        __pyx_t_23 = __Pyx_PyInt_As_GridCoord(__pyx_v_c); if (unlikely((__pyx_t_23 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 132, __pyx_L1_error)
+        __pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_v_cfg, __pyx_n_s_objects); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 132, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_11);
-        __pyx_t_10 = __Pyx_PyObject_GetAttrStr(__pyx_t_11, __pyx_n_s_altar); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 123, __pyx_L1_error)
+        __pyx_t_10 = __Pyx_PyObject_GetAttrStr(__pyx_t_11, __pyx_n_s_altar); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 132, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_10);
         __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
-        __pyx_t_24 = __pyx_convert_map_from_py_std_3a__3a_string__and_int(__pyx_t_10); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 123, __pyx_L1_error)
+        __pyx_t_24 = __pyx_convert_map_from_py_std_3a__3a_string__and_int(__pyx_t_10); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 132, __pyx_L1_error)
         __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
         try {
-          __pyx_t_27 = new Altar(__pyx_t_23, __pyx_t_22, __pyx_t_24);
+          __pyx_t_27 = new Altar(__pyx_t_22, __pyx_t_23, __pyx_t_24);
         } catch(...) {
           __Pyx_CppExn2PyErr();
-          __PYX_ERR(0, 123, __pyx_L1_error)
+          __PYX_ERR(0, 132, __pyx_L1_error)
         }
         __pyx_v_converter = __pyx_t_27;
 
-        /* "mettagrid/mettagrid.pyx":122
- *                 elif map[r,c] == "generator":
- *                     converter = new Generator(r, c, cfg.objects.generator)
+        /* "mettagrid/mettagrid.pyx":131
+ *                         m = "generator.red"
+ *                     converter = new Generator(r, c, cfg.objects[m])
  *                 elif map[r,c] == "altar":             # <<<<<<<<<<<<<<
  *                     converter = new Altar(r, c, cfg.objects.altar)
  *                 elif map[r,c] == "armory":
@@ -23493,53 +23731,53 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
         goto __pyx_L28;
       }
 
-      /* "mettagrid/mettagrid.pyx":124
+      /* "mettagrid/mettagrid.pyx":133
  *                 elif map[r,c] == "altar":
  *                     converter = new Altar(r, c, cfg.objects.altar)
  *                 elif map[r,c] == "armory":             # <<<<<<<<<<<<<<
  *                     converter = new Armory(r, c, cfg.objects.armory)
  *                 elif map[r,c] == "lasery":
  */
-      __pyx_t_10 = PyTuple_New(2); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 124, __pyx_L1_error)
+      __pyx_t_10 = PyTuple_New(2); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 133, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_10);
       __Pyx_INCREF(__pyx_v_r);
       __Pyx_GIVEREF(__pyx_v_r);
-      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 0, __pyx_v_r)) __PYX_ERR(0, 124, __pyx_L1_error);
+      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 0, __pyx_v_r)) __PYX_ERR(0, 133, __pyx_L1_error);
       __Pyx_INCREF(__pyx_v_c);
       __Pyx_GIVEREF(__pyx_v_c);
-      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 1, __pyx_v_c)) __PYX_ERR(0, 124, __pyx_L1_error);
-      __pyx_t_11 = __Pyx_PyObject_GetItem(__pyx_v_map, __pyx_t_10); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 124, __pyx_L1_error)
+      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 1, __pyx_v_c)) __PYX_ERR(0, 133, __pyx_L1_error);
+      __pyx_t_11 = __Pyx_PyObject_GetItem(__pyx_v_map, __pyx_t_10); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 133, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_11);
       __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
-      __pyx_t_6 = (__Pyx_PyUnicode_Equals(__pyx_t_11, __pyx_n_u_armory, Py_EQ)); if (unlikely((__pyx_t_6 < 0))) __PYX_ERR(0, 124, __pyx_L1_error)
+      __pyx_t_6 = (__Pyx_PyUnicode_Equals(__pyx_t_11, __pyx_n_u_armory, Py_EQ)); if (unlikely((__pyx_t_6 < 0))) __PYX_ERR(0, 133, __pyx_L1_error)
       __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
       if (__pyx_t_6) {
 
-        /* "mettagrid/mettagrid.pyx":125
+        /* "mettagrid/mettagrid.pyx":134
  *                     converter = new Altar(r, c, cfg.objects.altar)
  *                 elif map[r,c] == "armory":
  *                     converter = new Armory(r, c, cfg.objects.armory)             # <<<<<<<<<<<<<<
  *                 elif map[r,c] == "lasery":
  *                     converter = new Lasery(r, c, cfg.objects.lasery)
  */
-        __pyx_t_22 = __Pyx_PyInt_As_GridCoord(__pyx_v_r); if (unlikely((__pyx_t_22 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 125, __pyx_L1_error)
-        __pyx_t_23 = __Pyx_PyInt_As_GridCoord(__pyx_v_c); if (unlikely((__pyx_t_23 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 125, __pyx_L1_error)
-        __pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_v_cfg, __pyx_n_s_objects); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 125, __pyx_L1_error)
+        __pyx_t_23 = __Pyx_PyInt_As_GridCoord(__pyx_v_r); if (unlikely((__pyx_t_23 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 134, __pyx_L1_error)
+        __pyx_t_22 = __Pyx_PyInt_As_GridCoord(__pyx_v_c); if (unlikely((__pyx_t_22 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 134, __pyx_L1_error)
+        __pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_v_cfg, __pyx_n_s_objects); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 134, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_11);
-        __pyx_t_10 = __Pyx_PyObject_GetAttrStr(__pyx_t_11, __pyx_n_s_armory); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 125, __pyx_L1_error)
+        __pyx_t_10 = __Pyx_PyObject_GetAttrStr(__pyx_t_11, __pyx_n_s_armory); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 134, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_10);
         __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
-        __pyx_t_24 = __pyx_convert_map_from_py_std_3a__3a_string__and_int(__pyx_t_10); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 125, __pyx_L1_error)
+        __pyx_t_24 = __pyx_convert_map_from_py_std_3a__3a_string__and_int(__pyx_t_10); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 134, __pyx_L1_error)
         __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
         try {
-          __pyx_t_28 = new Armory(__pyx_t_22, __pyx_t_23, __pyx_t_24);
+          __pyx_t_28 = new Armory(__pyx_t_23, __pyx_t_22, __pyx_t_24);
         } catch(...) {
           __Pyx_CppExn2PyErr();
-          __PYX_ERR(0, 125, __pyx_L1_error)
+          __PYX_ERR(0, 134, __pyx_L1_error)
         }
         __pyx_v_converter = __pyx_t_28;
 
-        /* "mettagrid/mettagrid.pyx":124
+        /* "mettagrid/mettagrid.pyx":133
  *                 elif map[r,c] == "altar":
  *                     converter = new Altar(r, c, cfg.objects.altar)
  *                 elif map[r,c] == "armory":             # <<<<<<<<<<<<<<
@@ -23549,53 +23787,53 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
         goto __pyx_L28;
       }
 
-      /* "mettagrid/mettagrid.pyx":126
+      /* "mettagrid/mettagrid.pyx":135
  *                 elif map[r,c] == "armory":
  *                     converter = new Armory(r, c, cfg.objects.armory)
  *                 elif map[r,c] == "lasery":             # <<<<<<<<<<<<<<
  *                     converter = new Lasery(r, c, cfg.objects.lasery)
  *                 elif map[r,c] == "lab":
  */
-      __pyx_t_10 = PyTuple_New(2); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 126, __pyx_L1_error)
+      __pyx_t_10 = PyTuple_New(2); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 135, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_10);
       __Pyx_INCREF(__pyx_v_r);
       __Pyx_GIVEREF(__pyx_v_r);
-      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 0, __pyx_v_r)) __PYX_ERR(0, 126, __pyx_L1_error);
+      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 0, __pyx_v_r)) __PYX_ERR(0, 135, __pyx_L1_error);
       __Pyx_INCREF(__pyx_v_c);
       __Pyx_GIVEREF(__pyx_v_c);
-      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 1, __pyx_v_c)) __PYX_ERR(0, 126, __pyx_L1_error);
-      __pyx_t_11 = __Pyx_PyObject_GetItem(__pyx_v_map, __pyx_t_10); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 126, __pyx_L1_error)
+      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 1, __pyx_v_c)) __PYX_ERR(0, 135, __pyx_L1_error);
+      __pyx_t_11 = __Pyx_PyObject_GetItem(__pyx_v_map, __pyx_t_10); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 135, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_11);
       __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
-      __pyx_t_6 = (__Pyx_PyUnicode_Equals(__pyx_t_11, __pyx_n_u_lasery, Py_EQ)); if (unlikely((__pyx_t_6 < 0))) __PYX_ERR(0, 126, __pyx_L1_error)
+      __pyx_t_6 = (__Pyx_PyUnicode_Equals(__pyx_t_11, __pyx_n_u_lasery, Py_EQ)); if (unlikely((__pyx_t_6 < 0))) __PYX_ERR(0, 135, __pyx_L1_error)
       __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
       if (__pyx_t_6) {
 
-        /* "mettagrid/mettagrid.pyx":127
+        /* "mettagrid/mettagrid.pyx":136
  *                     converter = new Armory(r, c, cfg.objects.armory)
  *                 elif map[r,c] == "lasery":
  *                     converter = new Lasery(r, c, cfg.objects.lasery)             # <<<<<<<<<<<<<<
  *                 elif map[r,c] == "lab":
  *                     converter = new Lab(r, c, cfg.objects.lab)
  */
-        __pyx_t_23 = __Pyx_PyInt_As_GridCoord(__pyx_v_r); if (unlikely((__pyx_t_23 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 127, __pyx_L1_error)
-        __pyx_t_22 = __Pyx_PyInt_As_GridCoord(__pyx_v_c); if (unlikely((__pyx_t_22 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 127, __pyx_L1_error)
-        __pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_v_cfg, __pyx_n_s_objects); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 127, __pyx_L1_error)
+        __pyx_t_22 = __Pyx_PyInt_As_GridCoord(__pyx_v_r); if (unlikely((__pyx_t_22 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 136, __pyx_L1_error)
+        __pyx_t_23 = __Pyx_PyInt_As_GridCoord(__pyx_v_c); if (unlikely((__pyx_t_23 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 136, __pyx_L1_error)
+        __pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_v_cfg, __pyx_n_s_objects); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 136, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_11);
-        __pyx_t_10 = __Pyx_PyObject_GetAttrStr(__pyx_t_11, __pyx_n_s_lasery); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 127, __pyx_L1_error)
+        __pyx_t_10 = __Pyx_PyObject_GetAttrStr(__pyx_t_11, __pyx_n_s_lasery); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 136, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_10);
         __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
-        __pyx_t_24 = __pyx_convert_map_from_py_std_3a__3a_string__and_int(__pyx_t_10); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 127, __pyx_L1_error)
+        __pyx_t_24 = __pyx_convert_map_from_py_std_3a__3a_string__and_int(__pyx_t_10); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 136, __pyx_L1_error)
         __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
         try {
-          __pyx_t_29 = new Lasery(__pyx_t_23, __pyx_t_22, __pyx_t_24);
+          __pyx_t_29 = new Lasery(__pyx_t_22, __pyx_t_23, __pyx_t_24);
         } catch(...) {
           __Pyx_CppExn2PyErr();
-          __PYX_ERR(0, 127, __pyx_L1_error)
+          __PYX_ERR(0, 136, __pyx_L1_error)
         }
         __pyx_v_converter = __pyx_t_29;
 
-        /* "mettagrid/mettagrid.pyx":126
+        /* "mettagrid/mettagrid.pyx":135
  *                 elif map[r,c] == "armory":
  *                     converter = new Armory(r, c, cfg.objects.armory)
  *                 elif map[r,c] == "lasery":             # <<<<<<<<<<<<<<
@@ -23605,53 +23843,53 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
         goto __pyx_L28;
       }
 
-      /* "mettagrid/mettagrid.pyx":128
+      /* "mettagrid/mettagrid.pyx":137
  *                 elif map[r,c] == "lasery":
  *                     converter = new Lasery(r, c, cfg.objects.lasery)
  *                 elif map[r,c] == "lab":             # <<<<<<<<<<<<<<
  *                     converter = new Lab(r, c, cfg.objects.lab)
  *                 elif map[r,c] == "factory":
  */
-      __pyx_t_10 = PyTuple_New(2); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 128, __pyx_L1_error)
+      __pyx_t_10 = PyTuple_New(2); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 137, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_10);
       __Pyx_INCREF(__pyx_v_r);
       __Pyx_GIVEREF(__pyx_v_r);
-      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 0, __pyx_v_r)) __PYX_ERR(0, 128, __pyx_L1_error);
+      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 0, __pyx_v_r)) __PYX_ERR(0, 137, __pyx_L1_error);
       __Pyx_INCREF(__pyx_v_c);
       __Pyx_GIVEREF(__pyx_v_c);
-      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 1, __pyx_v_c)) __PYX_ERR(0, 128, __pyx_L1_error);
-      __pyx_t_11 = __Pyx_PyObject_GetItem(__pyx_v_map, __pyx_t_10); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 128, __pyx_L1_error)
+      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 1, __pyx_v_c)) __PYX_ERR(0, 137, __pyx_L1_error);
+      __pyx_t_11 = __Pyx_PyObject_GetItem(__pyx_v_map, __pyx_t_10); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 137, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_11);
       __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
-      __pyx_t_6 = (__Pyx_PyUnicode_Equals(__pyx_t_11, __pyx_n_u_lab, Py_EQ)); if (unlikely((__pyx_t_6 < 0))) __PYX_ERR(0, 128, __pyx_L1_error)
+      __pyx_t_6 = (__Pyx_PyUnicode_Equals(__pyx_t_11, __pyx_n_u_lab, Py_EQ)); if (unlikely((__pyx_t_6 < 0))) __PYX_ERR(0, 137, __pyx_L1_error)
       __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
       if (__pyx_t_6) {
 
-        /* "mettagrid/mettagrid.pyx":129
+        /* "mettagrid/mettagrid.pyx":138
  *                     converter = new Lasery(r, c, cfg.objects.lasery)
  *                 elif map[r,c] == "lab":
  *                     converter = new Lab(r, c, cfg.objects.lab)             # <<<<<<<<<<<<<<
  *                 elif map[r,c] == "factory":
  *                     converter = new Factory(r, c, cfg.objects.factory)
  */
-        __pyx_t_22 = __Pyx_PyInt_As_GridCoord(__pyx_v_r); if (unlikely((__pyx_t_22 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 129, __pyx_L1_error)
-        __pyx_t_23 = __Pyx_PyInt_As_GridCoord(__pyx_v_c); if (unlikely((__pyx_t_23 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 129, __pyx_L1_error)
-        __pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_v_cfg, __pyx_n_s_objects); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 129, __pyx_L1_error)
+        __pyx_t_23 = __Pyx_PyInt_As_GridCoord(__pyx_v_r); if (unlikely((__pyx_t_23 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 138, __pyx_L1_error)
+        __pyx_t_22 = __Pyx_PyInt_As_GridCoord(__pyx_v_c); if (unlikely((__pyx_t_22 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 138, __pyx_L1_error)
+        __pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_v_cfg, __pyx_n_s_objects); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 138, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_11);
-        __pyx_t_10 = __Pyx_PyObject_GetAttrStr(__pyx_t_11, __pyx_n_s_lab); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 129, __pyx_L1_error)
+        __pyx_t_10 = __Pyx_PyObject_GetAttrStr(__pyx_t_11, __pyx_n_s_lab); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 138, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_10);
         __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
-        __pyx_t_24 = __pyx_convert_map_from_py_std_3a__3a_string__and_int(__pyx_t_10); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 129, __pyx_L1_error)
+        __pyx_t_24 = __pyx_convert_map_from_py_std_3a__3a_string__and_int(__pyx_t_10); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 138, __pyx_L1_error)
         __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
         try {
-          __pyx_t_30 = new Lab(__pyx_t_22, __pyx_t_23, __pyx_t_24);
+          __pyx_t_30 = new Lab(__pyx_t_23, __pyx_t_22, __pyx_t_24);
         } catch(...) {
           __Pyx_CppExn2PyErr();
-          __PYX_ERR(0, 129, __pyx_L1_error)
+          __PYX_ERR(0, 138, __pyx_L1_error)
         }
         __pyx_v_converter = __pyx_t_30;
 
-        /* "mettagrid/mettagrid.pyx":128
+        /* "mettagrid/mettagrid.pyx":137
  *                 elif map[r,c] == "lasery":
  *                     converter = new Lasery(r, c, cfg.objects.lasery)
  *                 elif map[r,c] == "lab":             # <<<<<<<<<<<<<<
@@ -23661,53 +23899,53 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
         goto __pyx_L28;
       }
 
-      /* "mettagrid/mettagrid.pyx":130
+      /* "mettagrid/mettagrid.pyx":139
  *                 elif map[r,c] == "lab":
  *                     converter = new Lab(r, c, cfg.objects.lab)
  *                 elif map[r,c] == "factory":             # <<<<<<<<<<<<<<
  *                     converter = new Factory(r, c, cfg.objects.factory)
  *                 elif map[r,c] == "temple":
  */
-      __pyx_t_10 = PyTuple_New(2); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 130, __pyx_L1_error)
+      __pyx_t_10 = PyTuple_New(2); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 139, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_10);
       __Pyx_INCREF(__pyx_v_r);
       __Pyx_GIVEREF(__pyx_v_r);
-      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 0, __pyx_v_r)) __PYX_ERR(0, 130, __pyx_L1_error);
+      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 0, __pyx_v_r)) __PYX_ERR(0, 139, __pyx_L1_error);
       __Pyx_INCREF(__pyx_v_c);
       __Pyx_GIVEREF(__pyx_v_c);
-      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 1, __pyx_v_c)) __PYX_ERR(0, 130, __pyx_L1_error);
-      __pyx_t_11 = __Pyx_PyObject_GetItem(__pyx_v_map, __pyx_t_10); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 130, __pyx_L1_error)
+      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 1, __pyx_v_c)) __PYX_ERR(0, 139, __pyx_L1_error);
+      __pyx_t_11 = __Pyx_PyObject_GetItem(__pyx_v_map, __pyx_t_10); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 139, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_11);
       __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
-      __pyx_t_6 = (__Pyx_PyUnicode_Equals(__pyx_t_11, __pyx_n_u_factory, Py_EQ)); if (unlikely((__pyx_t_6 < 0))) __PYX_ERR(0, 130, __pyx_L1_error)
+      __pyx_t_6 = (__Pyx_PyUnicode_Equals(__pyx_t_11, __pyx_n_u_factory, Py_EQ)); if (unlikely((__pyx_t_6 < 0))) __PYX_ERR(0, 139, __pyx_L1_error)
       __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
       if (__pyx_t_6) {
 
-        /* "mettagrid/mettagrid.pyx":131
+        /* "mettagrid/mettagrid.pyx":140
  *                     converter = new Lab(r, c, cfg.objects.lab)
  *                 elif map[r,c] == "factory":
  *                     converter = new Factory(r, c, cfg.objects.factory)             # <<<<<<<<<<<<<<
  *                 elif map[r,c] == "temple":
  *                     converter = new Temple(r, c, cfg.objects.temple)
  */
-        __pyx_t_23 = __Pyx_PyInt_As_GridCoord(__pyx_v_r); if (unlikely((__pyx_t_23 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 131, __pyx_L1_error)
-        __pyx_t_22 = __Pyx_PyInt_As_GridCoord(__pyx_v_c); if (unlikely((__pyx_t_22 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 131, __pyx_L1_error)
-        __pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_v_cfg, __pyx_n_s_objects); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 131, __pyx_L1_error)
+        __pyx_t_22 = __Pyx_PyInt_As_GridCoord(__pyx_v_r); if (unlikely((__pyx_t_22 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 140, __pyx_L1_error)
+        __pyx_t_23 = __Pyx_PyInt_As_GridCoord(__pyx_v_c); if (unlikely((__pyx_t_23 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 140, __pyx_L1_error)
+        __pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_v_cfg, __pyx_n_s_objects); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 140, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_11);
-        __pyx_t_10 = __Pyx_PyObject_GetAttrStr(__pyx_t_11, __pyx_n_s_factory); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 131, __pyx_L1_error)
+        __pyx_t_10 = __Pyx_PyObject_GetAttrStr(__pyx_t_11, __pyx_n_s_factory); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 140, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_10);
         __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
-        __pyx_t_24 = __pyx_convert_map_from_py_std_3a__3a_string__and_int(__pyx_t_10); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 131, __pyx_L1_error)
+        __pyx_t_24 = __pyx_convert_map_from_py_std_3a__3a_string__and_int(__pyx_t_10); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 140, __pyx_L1_error)
         __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
         try {
-          __pyx_t_31 = new Factory(__pyx_t_23, __pyx_t_22, __pyx_t_24);
+          __pyx_t_31 = new Factory(__pyx_t_22, __pyx_t_23, __pyx_t_24);
         } catch(...) {
           __Pyx_CppExn2PyErr();
-          __PYX_ERR(0, 131, __pyx_L1_error)
+          __PYX_ERR(0, 140, __pyx_L1_error)
         }
         __pyx_v_converter = __pyx_t_31;
 
-        /* "mettagrid/mettagrid.pyx":130
+        /* "mettagrid/mettagrid.pyx":139
  *                 elif map[r,c] == "lab":
  *                     converter = new Lab(r, c, cfg.objects.lab)
  *                 elif map[r,c] == "factory":             # <<<<<<<<<<<<<<
@@ -23717,53 +23955,53 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
         goto __pyx_L28;
       }
 
-      /* "mettagrid/mettagrid.pyx":132
+      /* "mettagrid/mettagrid.pyx":141
  *                 elif map[r,c] == "factory":
  *                     converter = new Factory(r, c, cfg.objects.factory)
  *                 elif map[r,c] == "temple":             # <<<<<<<<<<<<<<
  *                     converter = new Temple(r, c, cfg.objects.temple)
  * 
  */
-      __pyx_t_10 = PyTuple_New(2); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 132, __pyx_L1_error)
+      __pyx_t_10 = PyTuple_New(2); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 141, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_10);
       __Pyx_INCREF(__pyx_v_r);
       __Pyx_GIVEREF(__pyx_v_r);
-      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 0, __pyx_v_r)) __PYX_ERR(0, 132, __pyx_L1_error);
+      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 0, __pyx_v_r)) __PYX_ERR(0, 141, __pyx_L1_error);
       __Pyx_INCREF(__pyx_v_c);
       __Pyx_GIVEREF(__pyx_v_c);
-      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 1, __pyx_v_c)) __PYX_ERR(0, 132, __pyx_L1_error);
-      __pyx_t_11 = __Pyx_PyObject_GetItem(__pyx_v_map, __pyx_t_10); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 132, __pyx_L1_error)
+      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_10, 1, __pyx_v_c)) __PYX_ERR(0, 141, __pyx_L1_error);
+      __pyx_t_11 = __Pyx_PyObject_GetItem(__pyx_v_map, __pyx_t_10); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 141, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_11);
       __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
-      __pyx_t_6 = (__Pyx_PyUnicode_Equals(__pyx_t_11, __pyx_n_u_temple, Py_EQ)); if (unlikely((__pyx_t_6 < 0))) __PYX_ERR(0, 132, __pyx_L1_error)
+      __pyx_t_6 = (__Pyx_PyUnicode_Equals(__pyx_t_11, __pyx_n_u_temple, Py_EQ)); if (unlikely((__pyx_t_6 < 0))) __PYX_ERR(0, 141, __pyx_L1_error)
       __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
       if (__pyx_t_6) {
 
-        /* "mettagrid/mettagrid.pyx":133
+        /* "mettagrid/mettagrid.pyx":142
  *                     converter = new Factory(r, c, cfg.objects.factory)
  *                 elif map[r,c] == "temple":
  *                     converter = new Temple(r, c, cfg.objects.temple)             # <<<<<<<<<<<<<<
  * 
  *                 elif map[r,c].startswith("agent."):
  */
-        __pyx_t_22 = __Pyx_PyInt_As_GridCoord(__pyx_v_r); if (unlikely((__pyx_t_22 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 133, __pyx_L1_error)
-        __pyx_t_23 = __Pyx_PyInt_As_GridCoord(__pyx_v_c); if (unlikely((__pyx_t_23 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 133, __pyx_L1_error)
-        __pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_v_cfg, __pyx_n_s_objects); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 133, __pyx_L1_error)
+        __pyx_t_23 = __Pyx_PyInt_As_GridCoord(__pyx_v_r); if (unlikely((__pyx_t_23 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 142, __pyx_L1_error)
+        __pyx_t_22 = __Pyx_PyInt_As_GridCoord(__pyx_v_c); if (unlikely((__pyx_t_22 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 142, __pyx_L1_error)
+        __pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_v_cfg, __pyx_n_s_objects); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 142, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_11);
-        __pyx_t_10 = __Pyx_PyObject_GetAttrStr(__pyx_t_11, __pyx_n_s_temple); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 133, __pyx_L1_error)
+        __pyx_t_10 = __Pyx_PyObject_GetAttrStr(__pyx_t_11, __pyx_n_s_temple); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 142, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_10);
         __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
-        __pyx_t_24 = __pyx_convert_map_from_py_std_3a__3a_string__and_int(__pyx_t_10); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 133, __pyx_L1_error)
+        __pyx_t_24 = __pyx_convert_map_from_py_std_3a__3a_string__and_int(__pyx_t_10); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 142, __pyx_L1_error)
         __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
         try {
-          __pyx_t_32 = new Temple(__pyx_t_22, __pyx_t_23, __pyx_t_24);
+          __pyx_t_32 = new Temple(__pyx_t_23, __pyx_t_22, __pyx_t_24);
         } catch(...) {
           __Pyx_CppExn2PyErr();
-          __PYX_ERR(0, 133, __pyx_L1_error)
+          __PYX_ERR(0, 142, __pyx_L1_error)
         }
         __pyx_v_converter = __pyx_t_32;
 
-        /* "mettagrid/mettagrid.pyx":132
+        /* "mettagrid/mettagrid.pyx":141
  *                 elif map[r,c] == "factory":
  *                     converter = new Factory(r, c, cfg.objects.factory)
  *                 elif map[r,c] == "temple":             # <<<<<<<<<<<<<<
@@ -23773,25 +24011,25 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
         goto __pyx_L28;
       }
 
-      /* "mettagrid/mettagrid.pyx":135
+      /* "mettagrid/mettagrid.pyx":144
  *                     converter = new Temple(r, c, cfg.objects.temple)
  * 
  *                 elif map[r,c].startswith("agent."):             # <<<<<<<<<<<<<<
  *                     group_name = map[r,c].split(".")[1]
  *                     agent_cfg = OmegaConf.to_container(OmegaConf.merge(
  */
-      __pyx_t_11 = PyTuple_New(2); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 135, __pyx_L1_error)
+      __pyx_t_11 = PyTuple_New(2); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 144, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_11);
       __Pyx_INCREF(__pyx_v_r);
       __Pyx_GIVEREF(__pyx_v_r);
-      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_11, 0, __pyx_v_r)) __PYX_ERR(0, 135, __pyx_L1_error);
+      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_11, 0, __pyx_v_r)) __PYX_ERR(0, 144, __pyx_L1_error);
       __Pyx_INCREF(__pyx_v_c);
       __Pyx_GIVEREF(__pyx_v_c);
-      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_11, 1, __pyx_v_c)) __PYX_ERR(0, 135, __pyx_L1_error);
-      __pyx_t_1 = __Pyx_PyObject_GetItem(__pyx_v_map, __pyx_t_11); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 135, __pyx_L1_error)
+      if (__Pyx_PyTuple_SET_ITEM(__pyx_t_11, 1, __pyx_v_c)) __PYX_ERR(0, 144, __pyx_L1_error);
+      __pyx_t_1 = __Pyx_PyObject_GetItem(__pyx_v_map, __pyx_t_11); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 144, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_1);
       __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
-      __pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_startswith); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 135, __pyx_L1_error)
+      __pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_startswith); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 144, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_11);
       __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
       __pyx_t_1 = NULL;
@@ -23812,33 +24050,33 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
         PyObject *__pyx_callargs[2] = {__pyx_t_1, __pyx_kp_u_agent};
         __pyx_t_10 = __Pyx_PyObject_FastCall(__pyx_t_11, __pyx_callargs+1-__pyx_t_5, 1+__pyx_t_5);
         __Pyx_XDECREF(__pyx_t_1); __pyx_t_1 = 0;
-        if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 135, __pyx_L1_error)
+        if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 144, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_10);
         __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
       }
-      __pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_10); if (unlikely((__pyx_t_6 < 0))) __PYX_ERR(0, 135, __pyx_L1_error)
+      __pyx_t_6 = __Pyx_PyObject_IsTrue(__pyx_t_10); if (unlikely((__pyx_t_6 < 0))) __PYX_ERR(0, 144, __pyx_L1_error)
       __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
       if (__pyx_t_6) {
 
-        /* "mettagrid/mettagrid.pyx":136
+        /* "mettagrid/mettagrid.pyx":145
  * 
  *                 elif map[r,c].startswith("agent."):
  *                     group_name = map[r,c].split(".")[1]             # <<<<<<<<<<<<<<
  *                     agent_cfg = OmegaConf.to_container(OmegaConf.merge(
  *                         cfg.agent, cfg.groups[group_name].props))
  */
-        __pyx_t_11 = PyTuple_New(2); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 136, __pyx_L1_error)
+        __pyx_t_11 = PyTuple_New(2); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 145, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_11);
         __Pyx_INCREF(__pyx_v_r);
         __Pyx_GIVEREF(__pyx_v_r);
-        if (__Pyx_PyTuple_SET_ITEM(__pyx_t_11, 0, __pyx_v_r)) __PYX_ERR(0, 136, __pyx_L1_error);
+        if (__Pyx_PyTuple_SET_ITEM(__pyx_t_11, 0, __pyx_v_r)) __PYX_ERR(0, 145, __pyx_L1_error);
         __Pyx_INCREF(__pyx_v_c);
         __Pyx_GIVEREF(__pyx_v_c);
-        if (__Pyx_PyTuple_SET_ITEM(__pyx_t_11, 1, __pyx_v_c)) __PYX_ERR(0, 136, __pyx_L1_error);
-        __pyx_t_1 = __Pyx_PyObject_GetItem(__pyx_v_map, __pyx_t_11); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 136, __pyx_L1_error)
+        if (__Pyx_PyTuple_SET_ITEM(__pyx_t_11, 1, __pyx_v_c)) __PYX_ERR(0, 145, __pyx_L1_error);
+        __pyx_t_1 = __Pyx_PyObject_GetItem(__pyx_v_map, __pyx_t_11); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 145, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_1);
         __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
-        __pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_split); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 136, __pyx_L1_error)
+        __pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_split); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 145, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_11);
         __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
         __pyx_t_1 = NULL;
@@ -23859,53 +24097,53 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
           PyObject *__pyx_callargs[2] = {__pyx_t_1, __pyx_kp_u__2};
           __pyx_t_10 = __Pyx_PyObject_FastCall(__pyx_t_11, __pyx_callargs+1-__pyx_t_5, 1+__pyx_t_5);
           __Pyx_XDECREF(__pyx_t_1); __pyx_t_1 = 0;
-          if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 136, __pyx_L1_error)
+          if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 145, __pyx_L1_error)
           __Pyx_GOTREF(__pyx_t_10);
           __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
         }
-        __pyx_t_11 = __Pyx_GetItemInt(__pyx_t_10, 1, long, 1, __Pyx_PyInt_From_long, 0, 0, 0); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 136, __pyx_L1_error)
+        __pyx_t_11 = __Pyx_GetItemInt(__pyx_t_10, 1, long, 1, __Pyx_PyInt_From_long, 0, 0, 0); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 145, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_11);
         __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
-        __pyx_t_33 = __pyx_convert_string_from_py_6libcpp_6string_std__in_string(__pyx_t_11); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 136, __pyx_L1_error)
+        __pyx_t_33 = __pyx_convert_string_from_py_6libcpp_6string_std__in_string(__pyx_t_11); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 145, __pyx_L1_error)
         __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
         __pyx_v_group_name = __PYX_STD_MOVE_IF_SUPPORTED(__pyx_t_33);
 
-        /* "mettagrid/mettagrid.pyx":137
+        /* "mettagrid/mettagrid.pyx":146
  *                 elif map[r,c].startswith("agent."):
  *                     group_name = map[r,c].split(".")[1]
  *                     agent_cfg = OmegaConf.to_container(OmegaConf.merge(             # <<<<<<<<<<<<<<
  *                         cfg.agent, cfg.groups[group_name].props))
  *                     rewards = agent_cfg.get("rewards", {})
  */
-        __Pyx_GetModuleGlobalName(__pyx_t_10, __pyx_n_s_OmegaConf); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 137, __pyx_L1_error)
+        __Pyx_GetModuleGlobalName(__pyx_t_10, __pyx_n_s_OmegaConf); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 146, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_10);
-        __pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_t_10, __pyx_n_s_to_container); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 137, __pyx_L1_error)
+        __pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_t_10, __pyx_n_s_to_container); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 146, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_1);
         __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
-        __Pyx_GetModuleGlobalName(__pyx_t_4, __pyx_n_s_OmegaConf); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 137, __pyx_L1_error)
+        __Pyx_GetModuleGlobalName(__pyx_t_4, __pyx_n_s_OmegaConf); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 146, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_4);
-        __pyx_t_8 = __Pyx_PyObject_GetAttrStr(__pyx_t_4, __pyx_n_s_merge); if (unlikely(!__pyx_t_8)) __PYX_ERR(0, 137, __pyx_L1_error)
+        __pyx_t_8 = __Pyx_PyObject_GetAttrStr(__pyx_t_4, __pyx_n_s_merge); if (unlikely(!__pyx_t_8)) __PYX_ERR(0, 146, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_8);
         __Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
 
-        /* "mettagrid/mettagrid.pyx":138
+        /* "mettagrid/mettagrid.pyx":147
  *                     group_name = map[r,c].split(".")[1]
  *                     agent_cfg = OmegaConf.to_container(OmegaConf.merge(
  *                         cfg.agent, cfg.groups[group_name].props))             # <<<<<<<<<<<<<<
  *                     rewards = agent_cfg.get("rewards", {})
  *                     del agent_cfg["rewards"]
  */
-        __pyx_t_4 = __Pyx_PyObject_GetAttrStr(__pyx_v_cfg, __pyx_n_s_agent_2); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 138, __pyx_L1_error)
+        __pyx_t_4 = __Pyx_PyObject_GetAttrStr(__pyx_v_cfg, __pyx_n_s_agent_2); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 147, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_4);
-        __pyx_t_2 = __Pyx_PyObject_GetAttrStr(__pyx_v_cfg, __pyx_n_s_groups); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 138, __pyx_L1_error)
+        __pyx_t_2 = __Pyx_PyObject_GetAttrStr(__pyx_v_cfg, __pyx_n_s_groups); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 147, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_2);
-        __pyx_t_3 = __pyx_convert_PyStr_string_to_py_6libcpp_6string_std__in_string(__pyx_v_group_name); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 138, __pyx_L1_error)
+        __pyx_t_3 = __pyx_convert_PyStr_string_to_py_6libcpp_6string_std__in_string(__pyx_v_group_name); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 147, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_3);
-        __pyx_t_34 = __Pyx_PyObject_Dict_GetItem(__pyx_t_2, __pyx_t_3); if (unlikely(!__pyx_t_34)) __PYX_ERR(0, 138, __pyx_L1_error)
+        __pyx_t_34 = __Pyx_PyObject_Dict_GetItem(__pyx_t_2, __pyx_t_3); if (unlikely(!__pyx_t_34)) __PYX_ERR(0, 147, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_34);
         __Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
         __Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
-        __pyx_t_3 = __Pyx_PyObject_GetAttrStr(__pyx_t_34, __pyx_n_s_props); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 138, __pyx_L1_error)
+        __pyx_t_3 = __Pyx_PyObject_GetAttrStr(__pyx_t_34, __pyx_n_s_props); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 147, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_3);
         __Pyx_DECREF(__pyx_t_34); __pyx_t_34 = 0;
         __pyx_t_34 = NULL;
@@ -23928,7 +24166,7 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
           __Pyx_XDECREF(__pyx_t_34); __pyx_t_34 = 0;
           __Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
           __Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
-          if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 137, __pyx_L1_error)
+          if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 146, __pyx_L1_error)
           __Pyx_GOTREF(__pyx_t_10);
           __Pyx_DECREF(__pyx_t_8); __pyx_t_8 = 0;
         }
@@ -23951,23 +24189,23 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
           __pyx_t_11 = __Pyx_PyObject_FastCall(__pyx_t_1, __pyx_callargs+1-__pyx_t_5, 1+__pyx_t_5);
           __Pyx_XDECREF(__pyx_t_8); __pyx_t_8 = 0;
           __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
-          if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 137, __pyx_L1_error)
+          if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 146, __pyx_L1_error)
           __Pyx_GOTREF(__pyx_t_11);
           __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
         }
         __Pyx_XDECREF_SET(__pyx_v_agent_cfg, __pyx_t_11);
         __pyx_t_11 = 0;
 
-        /* "mettagrid/mettagrid.pyx":139
+        /* "mettagrid/mettagrid.pyx":148
  *                     agent_cfg = OmegaConf.to_container(OmegaConf.merge(
  *                         cfg.agent, cfg.groups[group_name].props))
  *                     rewards = agent_cfg.get("rewards", {})             # <<<<<<<<<<<<<<
  *                     del agent_cfg["rewards"]
  *                     for inv_item in InventoryItemNames:
  */
-        __pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_agent_cfg, __pyx_n_s_get); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 139, __pyx_L1_error)
+        __pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_agent_cfg, __pyx_n_s_get); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 148, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_1);
-        __pyx_t_10 = __Pyx_PyDict_NewPresized(0); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 139, __pyx_L1_error)
+        __pyx_t_10 = __Pyx_PyDict_NewPresized(0); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 148, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_10);
         __pyx_t_8 = NULL;
         __pyx_t_5 = 0;
@@ -23988,23 +24226,23 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
           __pyx_t_11 = __Pyx_PyObject_FastCall(__pyx_t_1, __pyx_callargs+1-__pyx_t_5, 2+__pyx_t_5);
           __Pyx_XDECREF(__pyx_t_8); __pyx_t_8 = 0;
           __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
-          if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 139, __pyx_L1_error)
+          if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 148, __pyx_L1_error)
           __Pyx_GOTREF(__pyx_t_11);
           __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
         }
         __Pyx_XDECREF_SET(__pyx_v_rewards, __pyx_t_11);
         __pyx_t_11 = 0;
 
-        /* "mettagrid/mettagrid.pyx":140
+        /* "mettagrid/mettagrid.pyx":149
  *                         cfg.agent, cfg.groups[group_name].props))
  *                     rewards = agent_cfg.get("rewards", {})
  *                     del agent_cfg["rewards"]             # <<<<<<<<<<<<<<
  *                     for inv_item in InventoryItemNames:
  *                         rewards[inv_item] = rewards.get(inv_item, 0)
  */
-        if (unlikely((PyObject_DelItem(__pyx_v_agent_cfg, __pyx_n_u_rewards) < 0))) __PYX_ERR(0, 140, __pyx_L1_error)
+        if (unlikely((PyObject_DelItem(__pyx_v_agent_cfg, __pyx_n_u_rewards) < 0))) __PYX_ERR(0, 149, __pyx_L1_error)
 
-        /* "mettagrid/mettagrid.pyx":141
+        /* "mettagrid/mettagrid.pyx":150
  *                     rewards = agent_cfg.get("rewards", {})
  *                     del agent_cfg["rewards"]
  *                     for inv_item in InventoryItemNames:             # <<<<<<<<<<<<<<
@@ -24018,16 +24256,16 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
           ++__pyx_t_35;
           __pyx_v_inv_item = __PYX_STD_MOVE_IF_SUPPORTED(__pyx_t_33);
 
-          /* "mettagrid/mettagrid.pyx":142
+          /* "mettagrid/mettagrid.pyx":151
  *                     del agent_cfg["rewards"]
  *                     for inv_item in InventoryItemNames:
  *                         rewards[inv_item] = rewards.get(inv_item, 0)             # <<<<<<<<<<<<<<
  *                         rewards[inv_item + "_max"] = rewards.get(inv_item + "_max", 1000)
  *                     group_id = cfg.groups[group_name].id
  */
-          __pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_rewards, __pyx_n_s_get); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 142, __pyx_L1_error)
+          __pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_rewards, __pyx_n_s_get); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 151, __pyx_L1_error)
           __Pyx_GOTREF(__pyx_t_1);
-          __pyx_t_10 = __pyx_convert_PyStr_string_to_py_6libcpp_6string_std__in_string(__pyx_v_inv_item); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 142, __pyx_L1_error)
+          __pyx_t_10 = __pyx_convert_PyStr_string_to_py_6libcpp_6string_std__in_string(__pyx_v_inv_item); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 151, __pyx_L1_error)
           __Pyx_GOTREF(__pyx_t_10);
           __pyx_t_8 = NULL;
           __pyx_t_5 = 0;
@@ -24048,28 +24286,28 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
             __pyx_t_11 = __Pyx_PyObject_FastCall(__pyx_t_1, __pyx_callargs+1-__pyx_t_5, 2+__pyx_t_5);
             __Pyx_XDECREF(__pyx_t_8); __pyx_t_8 = 0;
             __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
-            if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 142, __pyx_L1_error)
+            if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 151, __pyx_L1_error)
             __Pyx_GOTREF(__pyx_t_11);
             __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
           }
-          __pyx_t_1 = __pyx_convert_PyStr_string_to_py_6libcpp_6string_std__in_string(__pyx_v_inv_item); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 142, __pyx_L1_error)
+          __pyx_t_1 = __pyx_convert_PyStr_string_to_py_6libcpp_6string_std__in_string(__pyx_v_inv_item); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 151, __pyx_L1_error)
           __Pyx_GOTREF(__pyx_t_1);
-          if (unlikely((PyObject_SetItem(__pyx_v_rewards, __pyx_t_1, __pyx_t_11) < 0))) __PYX_ERR(0, 142, __pyx_L1_error)
+          if (unlikely((PyObject_SetItem(__pyx_v_rewards, __pyx_t_1, __pyx_t_11) < 0))) __PYX_ERR(0, 151, __pyx_L1_error)
           __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
           __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
 
-          /* "mettagrid/mettagrid.pyx":143
+          /* "mettagrid/mettagrid.pyx":152
  *                     for inv_item in InventoryItemNames:
  *                         rewards[inv_item] = rewards.get(inv_item, 0)
  *                         rewards[inv_item + "_max"] = rewards.get(inv_item + "_max", 1000)             # <<<<<<<<<<<<<<
  *                     group_id = cfg.groups[group_name].id
  *                     agent = new Agent(
  */
-          __pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_rewards, __pyx_n_s_get); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 143, __pyx_L1_error)
+          __pyx_t_1 = __Pyx_PyObject_GetAttrStr(__pyx_v_rewards, __pyx_n_s_get); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 152, __pyx_L1_error)
           __Pyx_GOTREF(__pyx_t_1);
-          __pyx_t_10 = __pyx_convert_PyStr_string_to_py_6libcpp_6string_std__in_string(__pyx_v_inv_item); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 143, __pyx_L1_error)
+          __pyx_t_10 = __pyx_convert_PyStr_string_to_py_6libcpp_6string_std__in_string(__pyx_v_inv_item); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 152, __pyx_L1_error)
           __Pyx_GOTREF(__pyx_t_10);
-          __pyx_t_8 = __Pyx_PyUnicode_ConcatInPlace(__pyx_t_10, __pyx_n_u_max); if (unlikely(!__pyx_t_8)) __PYX_ERR(0, 143, __pyx_L1_error)
+          __pyx_t_8 = __Pyx_PyUnicode_ConcatInPlace(__pyx_t_10, __pyx_n_u_max); if (unlikely(!__pyx_t_8)) __PYX_ERR(0, 152, __pyx_L1_error)
           __Pyx_GOTREF(__pyx_t_8);
           __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
           __pyx_t_10 = NULL;
@@ -24091,20 +24329,20 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
             __pyx_t_11 = __Pyx_PyObject_FastCall(__pyx_t_1, __pyx_callargs+1-__pyx_t_5, 2+__pyx_t_5);
             __Pyx_XDECREF(__pyx_t_10); __pyx_t_10 = 0;
             __Pyx_DECREF(__pyx_t_8); __pyx_t_8 = 0;
-            if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 143, __pyx_L1_error)
+            if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 152, __pyx_L1_error)
             __Pyx_GOTREF(__pyx_t_11);
             __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
           }
-          __pyx_t_1 = __pyx_convert_PyStr_string_to_py_6libcpp_6string_std__in_string(__pyx_v_inv_item); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 143, __pyx_L1_error)
+          __pyx_t_1 = __pyx_convert_PyStr_string_to_py_6libcpp_6string_std__in_string(__pyx_v_inv_item); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 152, __pyx_L1_error)
           __Pyx_GOTREF(__pyx_t_1);
-          __pyx_t_8 = __Pyx_PyUnicode_ConcatInPlace(__pyx_t_1, __pyx_n_u_max); if (unlikely(!__pyx_t_8)) __PYX_ERR(0, 143, __pyx_L1_error)
+          __pyx_t_8 = __Pyx_PyUnicode_ConcatInPlace(__pyx_t_1, __pyx_n_u_max); if (unlikely(!__pyx_t_8)) __PYX_ERR(0, 152, __pyx_L1_error)
           __Pyx_GOTREF(__pyx_t_8);
           __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
-          if (unlikely((PyObject_SetItem(__pyx_v_rewards, __pyx_t_8, __pyx_t_11) < 0))) __PYX_ERR(0, 143, __pyx_L1_error)
+          if (unlikely((PyObject_SetItem(__pyx_v_rewards, __pyx_t_8, __pyx_t_11) < 0))) __PYX_ERR(0, 152, __pyx_L1_error)
           __Pyx_DECREF(__pyx_t_8); __pyx_t_8 = 0;
           __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
 
-          /* "mettagrid/mettagrid.pyx":141
+          /* "mettagrid/mettagrid.pyx":150
  *                     rewards = agent_cfg.get("rewards", {})
  *                     del agent_cfg["rewards"]
  *                     for inv_item in InventoryItemNames:             # <<<<<<<<<<<<<<
@@ -24113,50 +24351,50 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
  */
         }
 
-        /* "mettagrid/mettagrid.pyx":144
+        /* "mettagrid/mettagrid.pyx":153
  *                         rewards[inv_item] = rewards.get(inv_item, 0)
  *                         rewards[inv_item + "_max"] = rewards.get(inv_item + "_max", 1000)
  *                     group_id = cfg.groups[group_name].id             # <<<<<<<<<<<<<<
  *                     agent = new Agent(
  *                         r, c, group_name, group_id, agent_cfg, rewards)
  */
-        __pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_v_cfg, __pyx_n_s_groups); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 144, __pyx_L1_error)
+        __pyx_t_11 = __Pyx_PyObject_GetAttrStr(__pyx_v_cfg, __pyx_n_s_groups); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 153, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_11);
-        __pyx_t_8 = __pyx_convert_PyStr_string_to_py_6libcpp_6string_std__in_string(__pyx_v_group_name); if (unlikely(!__pyx_t_8)) __PYX_ERR(0, 144, __pyx_L1_error)
+        __pyx_t_8 = __pyx_convert_PyStr_string_to_py_6libcpp_6string_std__in_string(__pyx_v_group_name); if (unlikely(!__pyx_t_8)) __PYX_ERR(0, 153, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_8);
-        __pyx_t_1 = __Pyx_PyObject_Dict_GetItem(__pyx_t_11, __pyx_t_8); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 144, __pyx_L1_error)
+        __pyx_t_1 = __Pyx_PyObject_Dict_GetItem(__pyx_t_11, __pyx_t_8); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 153, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_1);
         __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
         __Pyx_DECREF(__pyx_t_8); __pyx_t_8 = 0;
-        __pyx_t_8 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_id); if (unlikely(!__pyx_t_8)) __PYX_ERR(0, 144, __pyx_L1_error)
+        __pyx_t_8 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_id); if (unlikely(!__pyx_t_8)) __PYX_ERR(0, 153, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_8);
         __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
-        __pyx_t_36 = __Pyx_PyInt_As_unsigned_char(__pyx_t_8); if (unlikely((__pyx_t_36 == (unsigned char)-1) && PyErr_Occurred())) __PYX_ERR(0, 144, __pyx_L1_error)
+        __pyx_t_36 = __Pyx_PyInt_As_unsigned_char(__pyx_t_8); if (unlikely((__pyx_t_36 == (unsigned char)-1) && PyErr_Occurred())) __PYX_ERR(0, 153, __pyx_L1_error)
         __Pyx_DECREF(__pyx_t_8); __pyx_t_8 = 0;
         __pyx_v_group_id = __pyx_t_36;
 
-        /* "mettagrid/mettagrid.pyx":146
+        /* "mettagrid/mettagrid.pyx":155
  *                     group_id = cfg.groups[group_name].id
  *                     agent = new Agent(
  *                         r, c, group_name, group_id, agent_cfg, rewards)             # <<<<<<<<<<<<<<
  *                     self._grid.add_object(agent)
  *                     agent.agent_id = self._agents.size()
  */
-        __pyx_t_23 = __Pyx_PyInt_As_GridCoord(__pyx_v_r); if (unlikely((__pyx_t_23 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 146, __pyx_L1_error)
-        __pyx_t_22 = __Pyx_PyInt_As_GridCoord(__pyx_v_c); if (unlikely((__pyx_t_22 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 146, __pyx_L1_error)
-        __pyx_t_24 = __pyx_convert_map_from_py_std_3a__3a_string__and_int(__pyx_v_agent_cfg); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 146, __pyx_L1_error)
-        __pyx_t_37 = __pyx_convert_map_from_py_std_3a__3a_string__and_float(__pyx_v_rewards); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 146, __pyx_L1_error)
+        __pyx_t_22 = __Pyx_PyInt_As_GridCoord(__pyx_v_r); if (unlikely((__pyx_t_22 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 155, __pyx_L1_error)
+        __pyx_t_23 = __Pyx_PyInt_As_GridCoord(__pyx_v_c); if (unlikely((__pyx_t_23 == ((GridCoord)-1)) && PyErr_Occurred())) __PYX_ERR(0, 155, __pyx_L1_error)
+        __pyx_t_24 = __pyx_convert_map_from_py_std_3a__3a_string__and_int(__pyx_v_agent_cfg); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 155, __pyx_L1_error)
+        __pyx_t_37 = __pyx_convert_map_from_py_std_3a__3a_string__and_float(__pyx_v_rewards); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 155, __pyx_L1_error)
 
-        /* "mettagrid/mettagrid.pyx":145
+        /* "mettagrid/mettagrid.pyx":154
  *                         rewards[inv_item + "_max"] = rewards.get(inv_item + "_max", 1000)
  *                     group_id = cfg.groups[group_name].id
  *                     agent = new Agent(             # <<<<<<<<<<<<<<
  *                         r, c, group_name, group_id, agent_cfg, rewards)
  *                     self._grid.add_object(agent)
  */
-        __pyx_v_agent = new Agent(__pyx_t_23, __pyx_t_22, __pyx_v_group_name, __pyx_v_group_id, __pyx_t_24, __pyx_t_37);
+        __pyx_v_agent = new Agent(__pyx_t_22, __pyx_t_23, __pyx_v_group_name, __pyx_v_group_id, __pyx_t_24, __pyx_t_37);
 
-        /* "mettagrid/mettagrid.pyx":147
+        /* "mettagrid/mettagrid.pyx":156
  *                     agent = new Agent(
  *                         r, c, group_name, group_id, agent_cfg, rewards)
  *                     self._grid.add_object(agent)             # <<<<<<<<<<<<<<
@@ -24165,7 +24403,7 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
  */
         (void)(__pyx_v_self->__pyx_base._grid->add_object(__pyx_v_agent));
 
-        /* "mettagrid/mettagrid.pyx":148
+        /* "mettagrid/mettagrid.pyx":157
  *                         r, c, group_name, group_id, agent_cfg, rewards)
  *                     self._grid.add_object(agent)
  *                     agent.agent_id = self._agents.size()             # <<<<<<<<<<<<<<
@@ -24174,16 +24412,16 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
  */
         __pyx_v_agent->agent_id = __pyx_v_self->__pyx_base._agents.size();
 
-        /* "mettagrid/mettagrid.pyx":149
+        /* "mettagrid/mettagrid.pyx":158
  *                     self._grid.add_object(agent)
  *                     agent.agent_id = self._agents.size()
  *                     self.add_agent(agent)             # <<<<<<<<<<<<<<
  *                     self._group_sizes[group_id] += 1
  * 
  */
-        ((struct __pyx_vtabstruct_9mettagrid_11mettagrid_c_MettaGrid *)__pyx_v_self->__pyx_base.__pyx_vtab)->__pyx_base.add_agent(((struct __pyx_obj_9mettagrid_8grid_env_GridEnv *)__pyx_v_self), __pyx_v_agent); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 149, __pyx_L1_error)
+        ((struct __pyx_vtabstruct_9mettagrid_11mettagrid_c_MettaGrid *)__pyx_v_self->__pyx_base.__pyx_vtab)->__pyx_base.add_agent(((struct __pyx_obj_9mettagrid_8grid_env_GridEnv *)__pyx_v_self), __pyx_v_agent); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 158, __pyx_L1_error)
 
-        /* "mettagrid/mettagrid.pyx":150
+        /* "mettagrid/mettagrid.pyx":159
  *                     agent.agent_id = self._agents.size()
  *                     self.add_agent(agent)
  *                     self._group_sizes[group_id] += 1             # <<<<<<<<<<<<<<
@@ -24193,7 +24431,7 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
         __pyx_t_36 = __pyx_v_group_id;
         (__pyx_v_self->_group_sizes[__pyx_t_36]) = ((__pyx_v_self->_group_sizes[__pyx_t_36]) + 1);
 
-        /* "mettagrid/mettagrid.pyx":135
+        /* "mettagrid/mettagrid.pyx":144
  *                     converter = new Temple(r, c, cfg.objects.temple)
  * 
  *                 elif map[r,c].startswith("agent."):             # <<<<<<<<<<<<<<
@@ -24203,7 +24441,7 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
       }
       __pyx_L28:;
 
-      /* "mettagrid/mettagrid.pyx":152
+      /* "mettagrid/mettagrid.pyx":161
  *                     self._group_sizes[group_id] += 1
  * 
  *                 if converter != NULL:             # <<<<<<<<<<<<<<
@@ -24213,41 +24451,41 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
       __pyx_t_6 = (__pyx_v_converter != NULL);
       if (__pyx_t_6) {
 
-        /* "mettagrid/mettagrid.pyx":153
+        /* "mettagrid/mettagrid.pyx":162
  * 
  *                 if converter != NULL:
  *                     stat = "objects." + map[r,c]             # <<<<<<<<<<<<<<
  *                     self._stats.incr(stat)
  *                     self._grid.add_object(converter)
  */
-        __pyx_t_8 = PyTuple_New(2); if (unlikely(!__pyx_t_8)) __PYX_ERR(0, 153, __pyx_L1_error)
+        __pyx_t_8 = PyTuple_New(2); if (unlikely(!__pyx_t_8)) __PYX_ERR(0, 162, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_8);
         __Pyx_INCREF(__pyx_v_r);
         __Pyx_GIVEREF(__pyx_v_r);
-        if (__Pyx_PyTuple_SET_ITEM(__pyx_t_8, 0, __pyx_v_r)) __PYX_ERR(0, 153, __pyx_L1_error);
+        if (__Pyx_PyTuple_SET_ITEM(__pyx_t_8, 0, __pyx_v_r)) __PYX_ERR(0, 162, __pyx_L1_error);
         __Pyx_INCREF(__pyx_v_c);
         __Pyx_GIVEREF(__pyx_v_c);
-        if (__Pyx_PyTuple_SET_ITEM(__pyx_t_8, 1, __pyx_v_c)) __PYX_ERR(0, 153, __pyx_L1_error);
-        __pyx_t_1 = __Pyx_PyObject_GetItem(__pyx_v_map, __pyx_t_8); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 153, __pyx_L1_error)
+        if (__Pyx_PyTuple_SET_ITEM(__pyx_t_8, 1, __pyx_v_c)) __PYX_ERR(0, 162, __pyx_L1_error);
+        __pyx_t_1 = __Pyx_PyObject_GetItem(__pyx_v_map, __pyx_t_8); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 162, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_1);
         __Pyx_DECREF(__pyx_t_8); __pyx_t_8 = 0;
-        __pyx_t_8 = PyNumber_Add(__pyx_kp_u_objects_2, __pyx_t_1); if (unlikely(!__pyx_t_8)) __PYX_ERR(0, 153, __pyx_L1_error)
+        __pyx_t_8 = PyNumber_Add(__pyx_kp_u_objects_2, __pyx_t_1); if (unlikely(!__pyx_t_8)) __PYX_ERR(0, 162, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_8);
         __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
         __Pyx_XDECREF_SET(__pyx_v_stat, __pyx_t_8);
         __pyx_t_8 = 0;
 
-        /* "mettagrid/mettagrid.pyx":154
+        /* "mettagrid/mettagrid.pyx":163
  *                 if converter != NULL:
  *                     stat = "objects." + map[r,c]
  *                     self._stats.incr(stat)             # <<<<<<<<<<<<<<
  *                     self._grid.add_object(converter)
  *                     converter.set_event_manager(&self._event_manager)
  */
-        __pyx_t_33 = __pyx_convert_string_from_py_6libcpp_6string_std__in_string(__pyx_v_stat); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 154, __pyx_L1_error)
+        __pyx_t_33 = __pyx_convert_string_from_py_6libcpp_6string_std__in_string(__pyx_v_stat); if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 163, __pyx_L1_error)
         __pyx_v_self->__pyx_base._stats.incr(__pyx_t_33);
 
-        /* "mettagrid/mettagrid.pyx":155
+        /* "mettagrid/mettagrid.pyx":164
  *                     stat = "objects." + map[r,c]
  *                     self._stats.incr(stat)
  *                     self._grid.add_object(converter)             # <<<<<<<<<<<<<<
@@ -24256,7 +24494,7 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
  */
         (void)(__pyx_v_self->__pyx_base._grid->add_object(__pyx_v_converter));
 
-        /* "mettagrid/mettagrid.pyx":156
+        /* "mettagrid/mettagrid.pyx":165
  *                     self._stats.incr(stat)
  *                     self._grid.add_object(converter)
  *                     converter.set_event_manager(&self._event_manager)             # <<<<<<<<<<<<<<
@@ -24265,7 +24503,7 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
  */
         __pyx_v_converter->set_event_manager((&__pyx_v_self->__pyx_base._event_manager));
 
-        /* "mettagrid/mettagrid.pyx":157
+        /* "mettagrid/mettagrid.pyx":166
  *                     self._grid.add_object(converter)
  *                     converter.set_event_manager(&self._event_manager)
  *                     converter = NULL             # <<<<<<<<<<<<<<
@@ -24274,7 +24512,7 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
  */
         __pyx_v_converter = NULL;
 
-        /* "mettagrid/mettagrid.pyx":152
+        /* "mettagrid/mettagrid.pyx":161
  *                     self._group_sizes[group_id] += 1
  * 
  *                 if converter != NULL:             # <<<<<<<<<<<<<<
@@ -24334,6 +24572,7 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
   __Pyx_XDECREF(__pyx_v_actions);
   __Pyx_XDECREF(__pyx_v_r);
   __Pyx_XDECREF(__pyx_v_c);
+  __Pyx_XDECREF(__pyx_v_m);
   __Pyx_XDECREF(__pyx_v_agent_cfg);
   __Pyx_XDECREF(__pyx_v_rewards);
   __Pyx_XDECREF(__pyx_v_stat);
@@ -24343,7 +24582,7 @@ static int __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid___init__(struct __pyx_ob
   return __pyx_r;
 }
 
-/* "mettagrid/mettagrid.pyx":161
+/* "mettagrid/mettagrid.pyx":170
  * 
  * 
  *     cpdef list[str] grid_features(self):             # <<<<<<<<<<<<<<
@@ -24379,7 +24618,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_features(struc
     if (unlikely(!__Pyx_object_dict_version_matches(((PyObject *)__pyx_v_self), __pyx_tp_dict_version, __pyx_obj_dict_version))) {
       PY_UINT64_T __pyx_typedict_guard = __Pyx_get_tp_dict_version(((PyObject *)__pyx_v_self));
       #endif
-      __pyx_t_1 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_grid_features); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 161, __pyx_L1_error)
+      __pyx_t_1 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_grid_features); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 170, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_1);
       if (!__Pyx_IsSameCFunction(__pyx_t_1, (void*) __pyx_pw_9mettagrid_11mettagrid_c_9MettaGrid_3grid_features)) {
         __Pyx_XDECREF(__pyx_r);
@@ -24402,11 +24641,11 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_features(struc
           PyObject *__pyx_callargs[2] = {__pyx_t_4, NULL};
           __pyx_t_2 = __Pyx_PyObject_FastCall(__pyx_t_3, __pyx_callargs+1-__pyx_t_5, 0+__pyx_t_5);
           __Pyx_XDECREF(__pyx_t_4); __pyx_t_4 = 0;
-          if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 161, __pyx_L1_error)
+          if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 170, __pyx_L1_error)
           __Pyx_GOTREF(__pyx_t_2);
           __Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
         }
-        if (!(likely(PyList_CheckExact(__pyx_t_2))||((__pyx_t_2) == Py_None) || __Pyx_RaiseUnexpectedTypeError("list", __pyx_t_2))) __PYX_ERR(0, 161, __pyx_L1_error)
+        if (!(likely(PyList_CheckExact(__pyx_t_2))||((__pyx_t_2) == Py_None) || __Pyx_RaiseUnexpectedTypeError("list", __pyx_t_2))) __PYX_ERR(0, 170, __pyx_L1_error)
         __pyx_r = ((PyObject*)__pyx_t_2);
         __pyx_t_2 = 0;
         __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
@@ -24425,7 +24664,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_features(struc
     #endif
   }
 
-  /* "mettagrid/mettagrid.pyx":162
+  /* "mettagrid/mettagrid.pyx":171
  * 
  *     cpdef list[str] grid_features(self):
  *         return self._grid_features             # <<<<<<<<<<<<<<
@@ -24433,14 +24672,14 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_features(struc
  *     def render(self):
  */
   __Pyx_XDECREF(__pyx_r);
-  __pyx_t_1 = __pyx_convert_vector_to_py_std_3a__3a_string(__pyx_v_self->__pyx_base._grid_features); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 162, __pyx_L1_error)
+  __pyx_t_1 = __pyx_convert_vector_to_py_std_3a__3a_string(__pyx_v_self->__pyx_base._grid_features); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 171, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_1);
-  if (!(likely(PyList_CheckExact(__pyx_t_1)) || __Pyx_RaiseUnexpectedTypeError("list", __pyx_t_1))) __PYX_ERR(0, 162, __pyx_L1_error)
+  if (!(likely(PyList_CheckExact(__pyx_t_1)) || __Pyx_RaiseUnexpectedTypeError("list", __pyx_t_1))) __PYX_ERR(0, 171, __pyx_L1_error)
   __pyx_r = ((PyObject*)__pyx_t_1);
   __pyx_t_1 = 0;
   goto __pyx_L0;
 
-  /* "mettagrid/mettagrid.pyx":161
+  /* "mettagrid/mettagrid.pyx":170
  * 
  * 
  *     cpdef list[str] grid_features(self):             # <<<<<<<<<<<<<<
@@ -24512,7 +24751,7 @@ static PyObject *__pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid_2grid_features(str
   int __pyx_clineno = 0;
   __Pyx_RefNannySetupContext("grid_features", 1);
   __Pyx_XDECREF(__pyx_r);
-  __pyx_t_1 = __pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_features(__pyx_v_self, 1); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 161, __pyx_L1_error)
+  __pyx_t_1 = __pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_features(__pyx_v_self, 1); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 170, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_1);
   __pyx_r = __pyx_t_1;
   __pyx_t_1 = 0;
@@ -24529,7 +24768,7 @@ static PyObject *__pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid_2grid_features(str
   return __pyx_r;
 }
 
-/* "mettagrid/mettagrid.pyx":164
+/* "mettagrid/mettagrid.pyx":173
  *         return self._grid_features
  * 
  *     def render(self):             # <<<<<<<<<<<<<<
@@ -24593,37 +24832,37 @@ static PyObject *__pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid_4render(struct __p
   int __pyx_clineno = 0;
   __Pyx_RefNannySetupContext("render", 1);
 
-  /* "mettagrid/mettagrid.pyx":165
+  /* "mettagrid/mettagrid.pyx":174
  * 
  *     def render(self):
  *         grid = self.render_ascii(["A", "#", "g", "c", "a"])             # <<<<<<<<<<<<<<
  *         for r in grid:
  *                 print("".join(r))
  */
-  __pyx_t_1 = PyList_New(5); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 165, __pyx_L1_error)
+  __pyx_t_1 = PyList_New(5); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 174, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_1);
   __Pyx_INCREF(__pyx_n_u_A);
   __Pyx_GIVEREF(__pyx_n_u_A);
-  if (__Pyx_PyList_SET_ITEM(__pyx_t_1, 0, __pyx_n_u_A)) __PYX_ERR(0, 165, __pyx_L1_error);
+  if (__Pyx_PyList_SET_ITEM(__pyx_t_1, 0, __pyx_n_u_A)) __PYX_ERR(0, 174, __pyx_L1_error);
   __Pyx_INCREF(__pyx_kp_u__12);
   __Pyx_GIVEREF(__pyx_kp_u__12);
-  if (__Pyx_PyList_SET_ITEM(__pyx_t_1, 1, __pyx_kp_u__12)) __PYX_ERR(0, 165, __pyx_L1_error);
+  if (__Pyx_PyList_SET_ITEM(__pyx_t_1, 1, __pyx_kp_u__12)) __PYX_ERR(0, 174, __pyx_L1_error);
   __Pyx_INCREF(__pyx_n_u_g);
   __Pyx_GIVEREF(__pyx_n_u_g);
-  if (__Pyx_PyList_SET_ITEM(__pyx_t_1, 2, __pyx_n_u_g)) __PYX_ERR(0, 165, __pyx_L1_error);
+  if (__Pyx_PyList_SET_ITEM(__pyx_t_1, 2, __pyx_n_u_g)) __PYX_ERR(0, 174, __pyx_L1_error);
   __Pyx_INCREF(__pyx_n_u_c);
   __Pyx_GIVEREF(__pyx_n_u_c);
-  if (__Pyx_PyList_SET_ITEM(__pyx_t_1, 3, __pyx_n_u_c)) __PYX_ERR(0, 165, __pyx_L1_error);
+  if (__Pyx_PyList_SET_ITEM(__pyx_t_1, 3, __pyx_n_u_c)) __PYX_ERR(0, 174, __pyx_L1_error);
   __Pyx_INCREF(__pyx_n_u_a);
   __Pyx_GIVEREF(__pyx_n_u_a);
-  if (__Pyx_PyList_SET_ITEM(__pyx_t_1, 4, __pyx_n_u_a)) __PYX_ERR(0, 165, __pyx_L1_error);
-  __pyx_t_2 = ((PyObject *)((struct __pyx_vtabstruct_9mettagrid_11mettagrid_c_MettaGrid *)__pyx_v_self->__pyx_base.__pyx_vtab)->__pyx_base.render_ascii(((struct __pyx_obj_9mettagrid_8grid_env_GridEnv *)__pyx_v_self), ((PyObject*)__pyx_t_1), 0)); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 165, __pyx_L1_error)
+  if (__Pyx_PyList_SET_ITEM(__pyx_t_1, 4, __pyx_n_u_a)) __PYX_ERR(0, 174, __pyx_L1_error);
+  __pyx_t_2 = ((PyObject *)((struct __pyx_vtabstruct_9mettagrid_11mettagrid_c_MettaGrid *)__pyx_v_self->__pyx_base.__pyx_vtab)->__pyx_base.render_ascii(((struct __pyx_obj_9mettagrid_8grid_env_GridEnv *)__pyx_v_self), ((PyObject*)__pyx_t_1), 0)); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 174, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_2);
   __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
   __pyx_v_grid = ((PyArrayObject *)__pyx_t_2);
   __pyx_t_2 = 0;
 
-  /* "mettagrid/mettagrid.pyx":166
+  /* "mettagrid/mettagrid.pyx":175
  *     def render(self):
  *         grid = self.render_ascii(["A", "#", "g", "c", "a"])
  *         for r in grid:             # <<<<<<<<<<<<<<
@@ -24635,9 +24874,9 @@ static PyObject *__pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid_4render(struct __p
     __pyx_t_3 = 0;
     __pyx_t_4 = NULL;
   } else {
-    __pyx_t_3 = -1; __pyx_t_2 = PyObject_GetIter(((PyObject *)__pyx_v_grid)); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 166, __pyx_L1_error)
+    __pyx_t_3 = -1; __pyx_t_2 = PyObject_GetIter(((PyObject *)__pyx_v_grid)); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 175, __pyx_L1_error)
     __Pyx_GOTREF(__pyx_t_2);
-    __pyx_t_4 = __Pyx_PyObject_GetIterNextFunc(__pyx_t_2); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 166, __pyx_L1_error)
+    __pyx_t_4 = __Pyx_PyObject_GetIterNextFunc(__pyx_t_2); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 175, __pyx_L1_error)
   }
   for (;;) {
     if (likely(!__pyx_t_4)) {
@@ -24645,28 +24884,28 @@ static PyObject *__pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid_4render(struct __p
         {
           Py_ssize_t __pyx_temp = __Pyx_PyList_GET_SIZE(__pyx_t_2);
           #if !CYTHON_ASSUME_SAFE_MACROS
-          if (unlikely((__pyx_temp < 0))) __PYX_ERR(0, 166, __pyx_L1_error)
+          if (unlikely((__pyx_temp < 0))) __PYX_ERR(0, 175, __pyx_L1_error)
           #endif
           if (__pyx_t_3 >= __pyx_temp) break;
         }
         #if CYTHON_ASSUME_SAFE_MACROS && !CYTHON_AVOID_BORROWED_REFS
-        __pyx_t_1 = PyList_GET_ITEM(__pyx_t_2, __pyx_t_3); __Pyx_INCREF(__pyx_t_1); __pyx_t_3++; if (unlikely((0 < 0))) __PYX_ERR(0, 166, __pyx_L1_error)
+        __pyx_t_1 = PyList_GET_ITEM(__pyx_t_2, __pyx_t_3); __Pyx_INCREF(__pyx_t_1); __pyx_t_3++; if (unlikely((0 < 0))) __PYX_ERR(0, 175, __pyx_L1_error)
         #else
-        __pyx_t_1 = __Pyx_PySequence_ITEM(__pyx_t_2, __pyx_t_3); __pyx_t_3++; if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 166, __pyx_L1_error)
+        __pyx_t_1 = __Pyx_PySequence_ITEM(__pyx_t_2, __pyx_t_3); __pyx_t_3++; if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 175, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_1);
         #endif
       } else {
         {
           Py_ssize_t __pyx_temp = __Pyx_PyTuple_GET_SIZE(__pyx_t_2);
           #if !CYTHON_ASSUME_SAFE_MACROS
-          if (unlikely((__pyx_temp < 0))) __PYX_ERR(0, 166, __pyx_L1_error)
+          if (unlikely((__pyx_temp < 0))) __PYX_ERR(0, 175, __pyx_L1_error)
           #endif
           if (__pyx_t_3 >= __pyx_temp) break;
         }
         #if CYTHON_ASSUME_SAFE_MACROS && !CYTHON_AVOID_BORROWED_REFS
-        __pyx_t_1 = PyTuple_GET_ITEM(__pyx_t_2, __pyx_t_3); __Pyx_INCREF(__pyx_t_1); __pyx_t_3++; if (unlikely((0 < 0))) __PYX_ERR(0, 166, __pyx_L1_error)
+        __pyx_t_1 = PyTuple_GET_ITEM(__pyx_t_2, __pyx_t_3); __Pyx_INCREF(__pyx_t_1); __pyx_t_3++; if (unlikely((0 < 0))) __PYX_ERR(0, 175, __pyx_L1_error)
         #else
-        __pyx_t_1 = __Pyx_PySequence_ITEM(__pyx_t_2, __pyx_t_3); __pyx_t_3++; if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 166, __pyx_L1_error)
+        __pyx_t_1 = __Pyx_PySequence_ITEM(__pyx_t_2, __pyx_t_3); __pyx_t_3++; if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 175, __pyx_L1_error)
         __Pyx_GOTREF(__pyx_t_1);
         #endif
       }
@@ -24676,7 +24915,7 @@ static PyObject *__pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid_4render(struct __p
         PyObject* exc_type = PyErr_Occurred();
         if (exc_type) {
           if (likely(__Pyx_PyErr_GivenExceptionMatches(exc_type, PyExc_StopIteration))) PyErr_Clear();
-          else __PYX_ERR(0, 166, __pyx_L1_error)
+          else __PYX_ERR(0, 175, __pyx_L1_error)
         }
         break;
       }
@@ -24685,21 +24924,21 @@ static PyObject *__pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid_4render(struct __p
     __Pyx_XDECREF_SET(__pyx_v_r, __pyx_t_1);
     __pyx_t_1 = 0;
 
-    /* "mettagrid/mettagrid.pyx":167
+    /* "mettagrid/mettagrid.pyx":176
  *         grid = self.render_ascii(["A", "#", "g", "c", "a"])
  *         for r in grid:
  *                 print("".join(r))             # <<<<<<<<<<<<<<
  * 
  *     cpdef grid_objects(self):
  */
-    __pyx_t_1 = PyUnicode_Join(__pyx_kp_u__13, __pyx_v_r); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 167, __pyx_L1_error)
+    __pyx_t_1 = PyUnicode_Join(__pyx_kp_u__13, __pyx_v_r); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 176, __pyx_L1_error)
     __Pyx_GOTREF(__pyx_t_1);
-    __pyx_t_5 = __Pyx_PyObject_CallOneArg(__pyx_builtin_print, __pyx_t_1); if (unlikely(!__pyx_t_5)) __PYX_ERR(0, 167, __pyx_L1_error)
+    __pyx_t_5 = __Pyx_PyObject_CallOneArg(__pyx_builtin_print, __pyx_t_1); if (unlikely(!__pyx_t_5)) __PYX_ERR(0, 176, __pyx_L1_error)
     __Pyx_GOTREF(__pyx_t_5);
     __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
     __Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
 
-    /* "mettagrid/mettagrid.pyx":166
+    /* "mettagrid/mettagrid.pyx":175
  *     def render(self):
  *         grid = self.render_ascii(["A", "#", "g", "c", "a"])
  *         for r in grid:             # <<<<<<<<<<<<<<
@@ -24709,7 +24948,7 @@ static PyObject *__pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid_4render(struct __p
   }
   __Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
 
-  /* "mettagrid/mettagrid.pyx":164
+  /* "mettagrid/mettagrid.pyx":173
  *         return self._grid_features
  * 
  *     def render(self):             # <<<<<<<<<<<<<<
@@ -24734,7 +24973,7 @@ static PyObject *__pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid_4render(struct __p
   return __pyx_r;
 }
 
-/* "mettagrid/mettagrid.pyx":169
+/* "mettagrid/mettagrid.pyx":178
  *                 print("".join(r))
  * 
  *     cpdef grid_objects(self):             # <<<<<<<<<<<<<<
@@ -24791,7 +25030,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_objects(struct
     if (unlikely(!__Pyx_object_dict_version_matches(((PyObject *)__pyx_v_self), __pyx_tp_dict_version, __pyx_obj_dict_version))) {
       PY_UINT64_T __pyx_typedict_guard = __Pyx_get_tp_dict_version(((PyObject *)__pyx_v_self));
       #endif
-      __pyx_t_1 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_grid_objects); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 169, __pyx_L1_error)
+      __pyx_t_1 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_grid_objects); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 178, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_1);
       if (!__Pyx_IsSameCFunction(__pyx_t_1, (void*) __pyx_pw_9mettagrid_11mettagrid_c_9MettaGrid_7grid_objects)) {
         __Pyx_XDECREF(__pyx_r);
@@ -24814,7 +25053,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_objects(struct
           PyObject *__pyx_callargs[2] = {__pyx_t_4, NULL};
           __pyx_t_2 = __Pyx_PyObject_FastCall(__pyx_t_3, __pyx_callargs+1-__pyx_t_5, 0+__pyx_t_5);
           __Pyx_XDECREF(__pyx_t_4); __pyx_t_4 = 0;
-          if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 169, __pyx_L1_error)
+          if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 178, __pyx_L1_error)
           __Pyx_GOTREF(__pyx_t_2);
           __Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
         }
@@ -24836,51 +25075,51 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_objects(struct
     #endif
   }
 
-  /* "mettagrid/mettagrid.pyx":171
+  /* "mettagrid/mettagrid.pyx":180
  *     cpdef grid_objects(self):
  *         cdef GridObject *obj
  *         cdef ObsType[:] obj_data = np.zeros(len(self.grid_features()), dtype=self._obs_encoder.obs_np_type())             # <<<<<<<<<<<<<<
  *         cdef unsigned int obj_id, i
  *         cdef ObservationEncoder obs_encoder = self._obs_encoder
  */
-  __Pyx_GetModuleGlobalName(__pyx_t_1, __pyx_n_s_np); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 171, __pyx_L1_error)
+  __Pyx_GetModuleGlobalName(__pyx_t_1, __pyx_n_s_np); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 180, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_1);
-  __pyx_t_2 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_zeros); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 171, __pyx_L1_error)
+  __pyx_t_2 = __Pyx_PyObject_GetAttrStr(__pyx_t_1, __pyx_n_s_zeros); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 180, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_2);
   __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
-  __pyx_t_1 = ((struct __pyx_vtabstruct_9mettagrid_11mettagrid_c_MettaGrid *)__pyx_v_self->__pyx_base.__pyx_vtab)->__pyx_base.grid_features(((struct __pyx_obj_9mettagrid_8grid_env_GridEnv *)__pyx_v_self), 0); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 171, __pyx_L1_error)
+  __pyx_t_1 = ((struct __pyx_vtabstruct_9mettagrid_11mettagrid_c_MettaGrid *)__pyx_v_self->__pyx_base.__pyx_vtab)->__pyx_base.grid_features(((struct __pyx_obj_9mettagrid_8grid_env_GridEnv *)__pyx_v_self), 0); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 180, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_1);
   if (unlikely(__pyx_t_1 == Py_None)) {
     PyErr_SetString(PyExc_TypeError, "object of type 'NoneType' has no len()");
-    __PYX_ERR(0, 171, __pyx_L1_error)
+    __PYX_ERR(0, 180, __pyx_L1_error)
   }
-  __pyx_t_6 = __Pyx_PyList_GET_SIZE(__pyx_t_1); if (unlikely(__pyx_t_6 == ((Py_ssize_t)-1))) __PYX_ERR(0, 171, __pyx_L1_error)
+  __pyx_t_6 = __Pyx_PyList_GET_SIZE(__pyx_t_1); if (unlikely(__pyx_t_6 == ((Py_ssize_t)-1))) __PYX_ERR(0, 180, __pyx_L1_error)
   __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
-  __pyx_t_1 = PyInt_FromSsize_t(__pyx_t_6); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 171, __pyx_L1_error)
+  __pyx_t_1 = PyInt_FromSsize_t(__pyx_t_6); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 180, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_1);
-  __pyx_t_3 = PyTuple_New(1); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 171, __pyx_L1_error)
+  __pyx_t_3 = PyTuple_New(1); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 180, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_3);
   __Pyx_GIVEREF(__pyx_t_1);
-  if (__Pyx_PyTuple_SET_ITEM(__pyx_t_3, 0, __pyx_t_1)) __PYX_ERR(0, 171, __pyx_L1_error);
+  if (__Pyx_PyTuple_SET_ITEM(__pyx_t_3, 0, __pyx_t_1)) __PYX_ERR(0, 180, __pyx_L1_error);
   __pyx_t_1 = 0;
-  __pyx_t_1 = __Pyx_PyDict_NewPresized(1); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 171, __pyx_L1_error)
+  __pyx_t_1 = __Pyx_PyDict_NewPresized(1); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 180, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_1);
-  __pyx_t_4 = ((struct __pyx_vtabstruct_9mettagrid_19observation_encoder_ObservationEncoder *)__pyx_v_self->__pyx_base._obs_encoder->__pyx_vtab)->obs_np_type(__pyx_v_self->__pyx_base._obs_encoder, 0); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 171, __pyx_L1_error)
+  __pyx_t_4 = ((struct __pyx_vtabstruct_9mettagrid_19observation_encoder_ObservationEncoder *)__pyx_v_self->__pyx_base._obs_encoder->__pyx_vtab)->obs_np_type(__pyx_v_self->__pyx_base._obs_encoder, 0); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 180, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_4);
-  if (PyDict_SetItem(__pyx_t_1, __pyx_n_s_dtype, __pyx_t_4) < 0) __PYX_ERR(0, 171, __pyx_L1_error)
+  if (PyDict_SetItem(__pyx_t_1, __pyx_n_s_dtype, __pyx_t_4) < 0) __PYX_ERR(0, 180, __pyx_L1_error)
   __Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
-  __pyx_t_4 = __Pyx_PyObject_Call(__pyx_t_2, __pyx_t_3, __pyx_t_1); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 171, __pyx_L1_error)
+  __pyx_t_4 = __Pyx_PyObject_Call(__pyx_t_2, __pyx_t_3, __pyx_t_1); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 180, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_4);
   __Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
   __Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
   __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
-  __pyx_t_7 = __Pyx_PyObject_to_MemoryviewSlice_ds_nn_ObsType(__pyx_t_4, PyBUF_WRITABLE); if (unlikely(!__pyx_t_7.memview)) __PYX_ERR(0, 171, __pyx_L1_error)
+  __pyx_t_7 = __Pyx_PyObject_to_MemoryviewSlice_ds_nn_ObsType(__pyx_t_4, PyBUF_WRITABLE); if (unlikely(!__pyx_t_7.memview)) __PYX_ERR(0, 180, __pyx_L1_error)
   __Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
   __pyx_v_obj_data = __pyx_t_7;
   __pyx_t_7.memview = NULL;
   __pyx_t_7.data = NULL;
 
-  /* "mettagrid/mettagrid.pyx":173
+  /* "mettagrid/mettagrid.pyx":182
  *         cdef ObsType[:] obj_data = np.zeros(len(self.grid_features()), dtype=self._obs_encoder.obs_np_type())
  *         cdef unsigned int obj_id, i
  *         cdef ObservationEncoder obs_encoder = self._obs_encoder             # <<<<<<<<<<<<<<
@@ -24892,19 +25131,19 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_objects(struct
   __pyx_v_obs_encoder = ((struct __pyx_obj_9mettagrid_19observation_encoder_ObservationEncoder *)__pyx_t_4);
   __pyx_t_4 = 0;
 
-  /* "mettagrid/mettagrid.pyx":175
+  /* "mettagrid/mettagrid.pyx":184
  *         cdef ObservationEncoder obs_encoder = self._obs_encoder
  *         cdef vector[unsigned int] offsets
  *         objects = {}             # <<<<<<<<<<<<<<
  *         for obj_id in range(1, self._grid.objects.size()):
  *             obj = self._grid.object(obj_id)
  */
-  __pyx_t_4 = __Pyx_PyDict_NewPresized(0); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 175, __pyx_L1_error)
+  __pyx_t_4 = __Pyx_PyDict_NewPresized(0); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 184, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_4);
   __pyx_v_objects = ((PyObject*)__pyx_t_4);
   __pyx_t_4 = 0;
 
-  /* "mettagrid/mettagrid.pyx":176
+  /* "mettagrid/mettagrid.pyx":185
  *         cdef vector[unsigned int] offsets
  *         objects = {}
  *         for obj_id in range(1, self._grid.objects.size()):             # <<<<<<<<<<<<<<
@@ -24916,7 +25155,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_objects(struct
   for (__pyx_t_5 = 1; __pyx_t_5 < __pyx_t_9; __pyx_t_5+=1) {
     __pyx_v_obj_id = __pyx_t_5;
 
-    /* "mettagrid/mettagrid.pyx":177
+    /* "mettagrid/mettagrid.pyx":186
  *         objects = {}
  *         for obj_id in range(1, self._grid.objects.size()):
  *             obj = self._grid.object(obj_id)             # <<<<<<<<<<<<<<
@@ -24925,7 +25164,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_objects(struct
  */
     __pyx_v_obj = __pyx_v_self->__pyx_base._grid->object(__pyx_v_obj_id);
 
-    /* "mettagrid/mettagrid.pyx":178
+    /* "mettagrid/mettagrid.pyx":187
  *         for obj_id in range(1, self._grid.objects.size()):
  *             obj = self._grid.object(obj_id)
  *             if obj == NULL:             # <<<<<<<<<<<<<<
@@ -24935,7 +25174,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_objects(struct
     __pyx_t_10 = (__pyx_v_obj == NULL);
     if (__pyx_t_10) {
 
-      /* "mettagrid/mettagrid.pyx":179
+      /* "mettagrid/mettagrid.pyx":188
  *             obj = self._grid.object(obj_id)
  *             if obj == NULL:
  *                 continue             # <<<<<<<<<<<<<<
@@ -24944,7 +25183,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_objects(struct
  */
       goto __pyx_L3_continue;
 
-      /* "mettagrid/mettagrid.pyx":178
+      /* "mettagrid/mettagrid.pyx":187
  *         for obj_id in range(1, self._grid.objects.size()):
  *             obj = self._grid.object(obj_id)
  *             if obj == NULL:             # <<<<<<<<<<<<<<
@@ -24953,82 +25192,82 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_objects(struct
  */
     }
 
-    /* "mettagrid/mettagrid.pyx":181
+    /* "mettagrid/mettagrid.pyx":190
  *                 continue
  *             objects[obj_id] = {
  *                 "id": obj_id,             # <<<<<<<<<<<<<<
  *                 "type": obj._type_id,
  *                 "r": obj.location.r,
  */
-    __pyx_t_4 = __Pyx_PyDict_NewPresized(5); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 181, __pyx_L1_error)
+    __pyx_t_4 = __Pyx_PyDict_NewPresized(5); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 190, __pyx_L1_error)
     __Pyx_GOTREF(__pyx_t_4);
-    __pyx_t_1 = __Pyx_PyInt_From_unsigned_int(__pyx_v_obj_id); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 181, __pyx_L1_error)
+    __pyx_t_1 = __Pyx_PyInt_From_unsigned_int(__pyx_v_obj_id); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 190, __pyx_L1_error)
     __Pyx_GOTREF(__pyx_t_1);
-    if (PyDict_SetItem(__pyx_t_4, __pyx_n_u_id, __pyx_t_1) < 0) __PYX_ERR(0, 181, __pyx_L1_error)
+    if (PyDict_SetItem(__pyx_t_4, __pyx_n_u_id, __pyx_t_1) < 0) __PYX_ERR(0, 190, __pyx_L1_error)
     __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
 
-    /* "mettagrid/mettagrid.pyx":182
+    /* "mettagrid/mettagrid.pyx":191
  *             objects[obj_id] = {
  *                 "id": obj_id,
  *                 "type": obj._type_id,             # <<<<<<<<<<<<<<
  *                 "r": obj.location.r,
  *                 "c": obj.location.c,
  */
-    __pyx_t_1 = __Pyx_PyInt_From_TypeId(__pyx_v_obj->_type_id); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 182, __pyx_L1_error)
+    __pyx_t_1 = __Pyx_PyInt_From_TypeId(__pyx_v_obj->_type_id); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 191, __pyx_L1_error)
     __Pyx_GOTREF(__pyx_t_1);
-    if (PyDict_SetItem(__pyx_t_4, __pyx_n_u_type, __pyx_t_1) < 0) __PYX_ERR(0, 181, __pyx_L1_error)
+    if (PyDict_SetItem(__pyx_t_4, __pyx_n_u_type, __pyx_t_1) < 0) __PYX_ERR(0, 190, __pyx_L1_error)
     __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
 
-    /* "mettagrid/mettagrid.pyx":183
+    /* "mettagrid/mettagrid.pyx":192
  *                 "id": obj_id,
  *                 "type": obj._type_id,
  *                 "r": obj.location.r,             # <<<<<<<<<<<<<<
  *                 "c": obj.location.c,
  *                 "layer": obj.location.layer
  */
-    __pyx_t_1 = __Pyx_PyInt_From_GridCoord(__pyx_v_obj->location.r); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 183, __pyx_L1_error)
+    __pyx_t_1 = __Pyx_PyInt_From_GridCoord(__pyx_v_obj->location.r); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 192, __pyx_L1_error)
     __Pyx_GOTREF(__pyx_t_1);
-    if (PyDict_SetItem(__pyx_t_4, __pyx_n_u_r, __pyx_t_1) < 0) __PYX_ERR(0, 181, __pyx_L1_error)
+    if (PyDict_SetItem(__pyx_t_4, __pyx_n_u_r, __pyx_t_1) < 0) __PYX_ERR(0, 190, __pyx_L1_error)
     __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
 
-    /* "mettagrid/mettagrid.pyx":184
+    /* "mettagrid/mettagrid.pyx":193
  *                 "type": obj._type_id,
  *                 "r": obj.location.r,
  *                 "c": obj.location.c,             # <<<<<<<<<<<<<<
  *                 "layer": obj.location.layer
  *             }
  */
-    __pyx_t_1 = __Pyx_PyInt_From_GridCoord(__pyx_v_obj->location.c); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 184, __pyx_L1_error)
+    __pyx_t_1 = __Pyx_PyInt_From_GridCoord(__pyx_v_obj->location.c); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 193, __pyx_L1_error)
     __Pyx_GOTREF(__pyx_t_1);
-    if (PyDict_SetItem(__pyx_t_4, __pyx_n_u_c, __pyx_t_1) < 0) __PYX_ERR(0, 181, __pyx_L1_error)
+    if (PyDict_SetItem(__pyx_t_4, __pyx_n_u_c, __pyx_t_1) < 0) __PYX_ERR(0, 190, __pyx_L1_error)
     __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
 
-    /* "mettagrid/mettagrid.pyx":185
+    /* "mettagrid/mettagrid.pyx":194
  *                 "r": obj.location.r,
  *                 "c": obj.location.c,
  *                 "layer": obj.location.layer             # <<<<<<<<<<<<<<
  *             }
  *             offsets.resize(obs_encoder._type_feature_names[obj._type_id].size())
  */
-    __pyx_t_1 = __Pyx_PyInt_From_Layer(__pyx_v_obj->location.layer); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 185, __pyx_L1_error)
+    __pyx_t_1 = __Pyx_PyInt_From_Layer(__pyx_v_obj->location.layer); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 194, __pyx_L1_error)
     __Pyx_GOTREF(__pyx_t_1);
-    if (PyDict_SetItem(__pyx_t_4, __pyx_n_u_layer, __pyx_t_1) < 0) __PYX_ERR(0, 181, __pyx_L1_error)
+    if (PyDict_SetItem(__pyx_t_4, __pyx_n_u_layer, __pyx_t_1) < 0) __PYX_ERR(0, 190, __pyx_L1_error)
     __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
 
-    /* "mettagrid/mettagrid.pyx":180
+    /* "mettagrid/mettagrid.pyx":189
  *             if obj == NULL:
  *                 continue
  *             objects[obj_id] = {             # <<<<<<<<<<<<<<
  *                 "id": obj_id,
  *                 "type": obj._type_id,
  */
-    __pyx_t_1 = __Pyx_PyInt_From_unsigned_int(__pyx_v_obj_id); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 180, __pyx_L1_error)
+    __pyx_t_1 = __Pyx_PyInt_From_unsigned_int(__pyx_v_obj_id); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 189, __pyx_L1_error)
     __Pyx_GOTREF(__pyx_t_1);
-    if (unlikely((PyDict_SetItem(__pyx_v_objects, __pyx_t_1, __pyx_t_4) < 0))) __PYX_ERR(0, 180, __pyx_L1_error)
+    if (unlikely((PyDict_SetItem(__pyx_v_objects, __pyx_t_1, __pyx_t_4) < 0))) __PYX_ERR(0, 189, __pyx_L1_error)
     __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
     __Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
 
-    /* "mettagrid/mettagrid.pyx":187
+    /* "mettagrid/mettagrid.pyx":196
  *                 "layer": obj.location.layer
  *             }
  *             offsets.resize(obs_encoder._type_feature_names[obj._type_id].size())             # <<<<<<<<<<<<<<
@@ -25039,10 +25278,10 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_objects(struct
       __pyx_v_offsets.resize((__pyx_v_obs_encoder->_type_feature_names[__pyx_v_obj->_type_id]).size());
     } catch(...) {
       __Pyx_CppExn2PyErr();
-      __PYX_ERR(0, 187, __pyx_L1_error)
+      __PYX_ERR(0, 196, __pyx_L1_error)
     }
 
-    /* "mettagrid/mettagrid.pyx":188
+    /* "mettagrid/mettagrid.pyx":197
  *             }
  *             offsets.resize(obs_encoder._type_feature_names[obj._type_id].size())
  *             for i in range(offsets.size()):             # <<<<<<<<<<<<<<
@@ -25054,7 +25293,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_objects(struct
     for (__pyx_t_13 = 0; __pyx_t_13 < __pyx_t_12; __pyx_t_13+=1) {
       __pyx_v_i = __pyx_t_13;
 
-      /* "mettagrid/mettagrid.pyx":189
+      /* "mettagrid/mettagrid.pyx":198
  *             offsets.resize(obs_encoder._type_feature_names[obj._type_id].size())
  *             for i in range(offsets.size()):
  *                 offsets[i] = i             # <<<<<<<<<<<<<<
@@ -25064,18 +25303,18 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_objects(struct
       (__pyx_v_offsets[__pyx_v_i]) = __pyx_v_i;
     }
 
-    /* "mettagrid/mettagrid.pyx":190
+    /* "mettagrid/mettagrid.pyx":199
  *             for i in range(offsets.size()):
  *                 offsets[i] = i
  *             obs_encoder._encode(obj, obj_data, offsets)             # <<<<<<<<<<<<<<
  *             for i, name in enumerate(obs_encoder._type_feature_names[obj._type_id]):
  *                 objects[obj_id][name] = obj_data[i]
  */
-    __pyx_t_4 = ((struct __pyx_vtabstruct_9mettagrid_19observation_encoder_ObservationEncoder *)__pyx_v_obs_encoder->__pyx_vtab)->_encode(__pyx_v_obs_encoder, __pyx_v_obj, __pyx_v_obj_data, __pyx_v_offsets); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 190, __pyx_L1_error)
+    __pyx_t_4 = ((struct __pyx_vtabstruct_9mettagrid_19observation_encoder_ObservationEncoder *)__pyx_v_obs_encoder->__pyx_vtab)->_encode(__pyx_v_obs_encoder, __pyx_v_obj, __pyx_v_obj_data, __pyx_v_offsets); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 199, __pyx_L1_error)
     __Pyx_GOTREF(__pyx_t_4);
     __Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
 
-    /* "mettagrid/mettagrid.pyx":191
+    /* "mettagrid/mettagrid.pyx":200
  *                 offsets[i] = i
  *             obs_encoder._encode(obj, obj_data, offsets)
  *             for i, name in enumerate(obs_encoder._type_feature_names[obj._type_id]):             # <<<<<<<<<<<<<<
@@ -25083,16 +25322,16 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_objects(struct
  * 
  */
     __pyx_t_13 = 0;
-    __pyx_t_4 = __pyx_convert_vector_to_py_std_3a__3a_string((__pyx_v_obs_encoder->_type_feature_names[__pyx_v_obj->_type_id])); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 191, __pyx_L1_error)
+    __pyx_t_4 = __pyx_convert_vector_to_py_std_3a__3a_string((__pyx_v_obs_encoder->_type_feature_names[__pyx_v_obj->_type_id])); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 200, __pyx_L1_error)
     __Pyx_GOTREF(__pyx_t_4);
     if (likely(PyList_CheckExact(__pyx_t_4)) || PyTuple_CheckExact(__pyx_t_4)) {
       __pyx_t_1 = __pyx_t_4; __Pyx_INCREF(__pyx_t_1);
       __pyx_t_6 = 0;
       __pyx_t_14 = NULL;
     } else {
-      __pyx_t_6 = -1; __pyx_t_1 = PyObject_GetIter(__pyx_t_4); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 191, __pyx_L1_error)
+      __pyx_t_6 = -1; __pyx_t_1 = PyObject_GetIter(__pyx_t_4); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 200, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_1);
-      __pyx_t_14 = __Pyx_PyObject_GetIterNextFunc(__pyx_t_1); if (unlikely(!__pyx_t_14)) __PYX_ERR(0, 191, __pyx_L1_error)
+      __pyx_t_14 = __Pyx_PyObject_GetIterNextFunc(__pyx_t_1); if (unlikely(!__pyx_t_14)) __PYX_ERR(0, 200, __pyx_L1_error)
     }
     __Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
     for (;;) {
@@ -25101,28 +25340,28 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_objects(struct
           {
             Py_ssize_t __pyx_temp = __Pyx_PyList_GET_SIZE(__pyx_t_1);
             #if !CYTHON_ASSUME_SAFE_MACROS
-            if (unlikely((__pyx_temp < 0))) __PYX_ERR(0, 191, __pyx_L1_error)
+            if (unlikely((__pyx_temp < 0))) __PYX_ERR(0, 200, __pyx_L1_error)
             #endif
             if (__pyx_t_6 >= __pyx_temp) break;
           }
           #if CYTHON_ASSUME_SAFE_MACROS && !CYTHON_AVOID_BORROWED_REFS
-          __pyx_t_4 = PyList_GET_ITEM(__pyx_t_1, __pyx_t_6); __Pyx_INCREF(__pyx_t_4); __pyx_t_6++; if (unlikely((0 < 0))) __PYX_ERR(0, 191, __pyx_L1_error)
+          __pyx_t_4 = PyList_GET_ITEM(__pyx_t_1, __pyx_t_6); __Pyx_INCREF(__pyx_t_4); __pyx_t_6++; if (unlikely((0 < 0))) __PYX_ERR(0, 200, __pyx_L1_error)
           #else
-          __pyx_t_4 = __Pyx_PySequence_ITEM(__pyx_t_1, __pyx_t_6); __pyx_t_6++; if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 191, __pyx_L1_error)
+          __pyx_t_4 = __Pyx_PySequence_ITEM(__pyx_t_1, __pyx_t_6); __pyx_t_6++; if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 200, __pyx_L1_error)
           __Pyx_GOTREF(__pyx_t_4);
           #endif
         } else {
           {
             Py_ssize_t __pyx_temp = __Pyx_PyTuple_GET_SIZE(__pyx_t_1);
             #if !CYTHON_ASSUME_SAFE_MACROS
-            if (unlikely((__pyx_temp < 0))) __PYX_ERR(0, 191, __pyx_L1_error)
+            if (unlikely((__pyx_temp < 0))) __PYX_ERR(0, 200, __pyx_L1_error)
             #endif
             if (__pyx_t_6 >= __pyx_temp) break;
           }
           #if CYTHON_ASSUME_SAFE_MACROS && !CYTHON_AVOID_BORROWED_REFS
-          __pyx_t_4 = PyTuple_GET_ITEM(__pyx_t_1, __pyx_t_6); __Pyx_INCREF(__pyx_t_4); __pyx_t_6++; if (unlikely((0 < 0))) __PYX_ERR(0, 191, __pyx_L1_error)
+          __pyx_t_4 = PyTuple_GET_ITEM(__pyx_t_1, __pyx_t_6); __Pyx_INCREF(__pyx_t_4); __pyx_t_6++; if (unlikely((0 < 0))) __PYX_ERR(0, 200, __pyx_L1_error)
           #else
-          __pyx_t_4 = __Pyx_PySequence_ITEM(__pyx_t_1, __pyx_t_6); __pyx_t_6++; if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 191, __pyx_L1_error)
+          __pyx_t_4 = __Pyx_PySequence_ITEM(__pyx_t_1, __pyx_t_6); __pyx_t_6++; if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 200, __pyx_L1_error)
           __Pyx_GOTREF(__pyx_t_4);
           #endif
         }
@@ -25132,7 +25371,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_objects(struct
           PyObject* exc_type = PyErr_Occurred();
           if (exc_type) {
             if (likely(__Pyx_PyErr_GivenExceptionMatches(exc_type, PyExc_StopIteration))) PyErr_Clear();
-            else __PYX_ERR(0, 191, __pyx_L1_error)
+            else __PYX_ERR(0, 200, __pyx_L1_error)
           }
           break;
         }
@@ -25143,7 +25382,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_objects(struct
       __pyx_v_i = __pyx_t_13;
       __pyx_t_13 = (__pyx_t_13 + 1);
 
-      /* "mettagrid/mettagrid.pyx":192
+      /* "mettagrid/mettagrid.pyx":201
  *             obs_encoder._encode(obj, obj_data, offsets)
  *             for i, name in enumerate(obs_encoder._type_feature_names[obj._type_id]):
  *                 objects[obj_id][name] = obj_data[i]             # <<<<<<<<<<<<<<
@@ -25151,18 +25390,18 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_objects(struct
  *         for agent_idx in range(self._agents.size()):
  */
       __pyx_t_15 = __pyx_v_i;
-      __pyx_t_4 = __Pyx_PyInt_From_ObsType((*((ObsType *) ( /* dim=0 */ (__pyx_v_obj_data.data + __pyx_t_15 * __pyx_v_obj_data.strides[0]) )))); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 192, __pyx_L1_error)
+      __pyx_t_4 = __Pyx_PyInt_From_ObsType((*((ObsType *) ( /* dim=0 */ (__pyx_v_obj_data.data + __pyx_t_15 * __pyx_v_obj_data.strides[0]) )))); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 201, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_4);
-      __pyx_t_3 = __Pyx_PyInt_From_unsigned_int(__pyx_v_obj_id); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 192, __pyx_L1_error)
+      __pyx_t_3 = __Pyx_PyInt_From_unsigned_int(__pyx_v_obj_id); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 201, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_3);
-      __pyx_t_2 = __Pyx_PyDict_GetItem(__pyx_v_objects, __pyx_t_3); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 192, __pyx_L1_error)
+      __pyx_t_2 = __Pyx_PyDict_GetItem(__pyx_v_objects, __pyx_t_3); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 201, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_2);
       __Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
-      if (unlikely((PyObject_SetItem(__pyx_t_2, __pyx_v_name, __pyx_t_4) < 0))) __PYX_ERR(0, 192, __pyx_L1_error)
+      if (unlikely((PyObject_SetItem(__pyx_t_2, __pyx_v_name, __pyx_t_4) < 0))) __PYX_ERR(0, 201, __pyx_L1_error)
       __Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
       __Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
 
-      /* "mettagrid/mettagrid.pyx":191
+      /* "mettagrid/mettagrid.pyx":200
  *                 offsets[i] = i
  *             obs_encoder._encode(obj, obj_data, offsets)
  *             for i, name in enumerate(obs_encoder._type_feature_names[obj._type_id]):             # <<<<<<<<<<<<<<
@@ -25174,7 +25413,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_objects(struct
     __pyx_L3_continue:;
   }
 
-  /* "mettagrid/mettagrid.pyx":194
+  /* "mettagrid/mettagrid.pyx":203
  *                 objects[obj_id][name] = obj_data[i]
  * 
  *         for agent_idx in range(self._agents.size()):             # <<<<<<<<<<<<<<
@@ -25186,35 +25425,35 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_objects(struct
   for (__pyx_t_16 = 0; __pyx_t_16 < __pyx_t_9; __pyx_t_16+=1) {
     __pyx_v_agent_idx = __pyx_t_16;
 
-    /* "mettagrid/mettagrid.pyx":195
+    /* "mettagrid/mettagrid.pyx":204
  * 
  *         for agent_idx in range(self._agents.size()):
  *             agent_object = objects[self._agents[agent_idx].id]             # <<<<<<<<<<<<<<
  *             agent_object["agent_id"] = agent_idx
  * 
  */
-    __pyx_t_1 = __Pyx_PyInt_From_GridObjectId((__pyx_v_self->__pyx_base._agents[__pyx_v_agent_idx])->id); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 195, __pyx_L1_error)
+    __pyx_t_1 = __Pyx_PyInt_From_GridObjectId((__pyx_v_self->__pyx_base._agents[__pyx_v_agent_idx])->id); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 204, __pyx_L1_error)
     __Pyx_GOTREF(__pyx_t_1);
-    __pyx_t_4 = __Pyx_PyDict_GetItem(__pyx_v_objects, __pyx_t_1); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 195, __pyx_L1_error)
+    __pyx_t_4 = __Pyx_PyDict_GetItem(__pyx_v_objects, __pyx_t_1); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 204, __pyx_L1_error)
     __Pyx_GOTREF(__pyx_t_4);
     __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
     __Pyx_XDECREF_SET(__pyx_v_agent_object, __pyx_t_4);
     __pyx_t_4 = 0;
 
-    /* "mettagrid/mettagrid.pyx":196
+    /* "mettagrid/mettagrid.pyx":205
  *         for agent_idx in range(self._agents.size()):
  *             agent_object = objects[self._agents[agent_idx].id]
  *             agent_object["agent_id"] = agent_idx             # <<<<<<<<<<<<<<
  * 
  *         return objects
  */
-    __pyx_t_4 = __Pyx_PyInt_FromSize_t(__pyx_v_agent_idx); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 196, __pyx_L1_error)
+    __pyx_t_4 = __Pyx_PyInt_FromSize_t(__pyx_v_agent_idx); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 205, __pyx_L1_error)
     __Pyx_GOTREF(__pyx_t_4);
-    if (unlikely((PyObject_SetItem(__pyx_v_agent_object, __pyx_n_u_agent_id, __pyx_t_4) < 0))) __PYX_ERR(0, 196, __pyx_L1_error)
+    if (unlikely((PyObject_SetItem(__pyx_v_agent_object, __pyx_n_u_agent_id, __pyx_t_4) < 0))) __PYX_ERR(0, 205, __pyx_L1_error)
     __Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
   }
 
-  /* "mettagrid/mettagrid.pyx":198
+  /* "mettagrid/mettagrid.pyx":207
  *             agent_object["agent_id"] = agent_idx
  * 
  *         return objects             # <<<<<<<<<<<<<<
@@ -25226,7 +25465,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_objects(struct
   __pyx_r = __pyx_v_objects;
   goto __pyx_L0;
 
-  /* "mettagrid/mettagrid.pyx":169
+  /* "mettagrid/mettagrid.pyx":178
  *                 print("".join(r))
  * 
  *     cpdef grid_objects(self):             # <<<<<<<<<<<<<<
@@ -25304,7 +25543,7 @@ static PyObject *__pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid_6grid_objects(stru
   int __pyx_clineno = 0;
   __Pyx_RefNannySetupContext("grid_objects", 1);
   __Pyx_XDECREF(__pyx_r);
-  __pyx_t_1 = __pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_objects(__pyx_v_self, 1); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 169, __pyx_L1_error)
+  __pyx_t_1 = __pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_grid_objects(__pyx_v_self, 1); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 178, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_1);
   __pyx_r = __pyx_t_1;
   __pyx_t_1 = 0;
@@ -25321,7 +25560,7 @@ static PyObject *__pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid_6grid_objects(stru
   return __pyx_r;
 }
 
-/* "mettagrid/mettagrid.pyx":200
+/* "mettagrid/mettagrid.pyx":209
  *         return objects
  * 
  *     cpdef tuple[cnp.ndarray, cnp.ndarray, cnp.ndarray, cnp.ndarray, dict] step(self, cnp.ndarray actions):             # <<<<<<<<<<<<<<
@@ -25379,7 +25618,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
     if (unlikely(!__Pyx_object_dict_version_matches(((PyObject *)__pyx_v_self), __pyx_tp_dict_version, __pyx_obj_dict_version))) {
       PY_UINT64_T __pyx_typedict_guard = __Pyx_get_tp_dict_version(((PyObject *)__pyx_v_self));
       #endif
-      __pyx_t_1 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_step); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 200, __pyx_L1_error)
+      __pyx_t_1 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_step); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 209, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_1);
       if (!__Pyx_IsSameCFunction(__pyx_t_1, (void*) __pyx_pw_9mettagrid_11mettagrid_c_9MettaGrid_9step)) {
         __Pyx_XDECREF(__pyx_r);
@@ -25402,11 +25641,11 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
           PyObject *__pyx_callargs[2] = {__pyx_t_4, ((PyObject *)__pyx_v_actions)};
           __pyx_t_2 = __Pyx_PyObject_FastCall(__pyx_t_3, __pyx_callargs+1-__pyx_t_5, 1+__pyx_t_5);
           __Pyx_XDECREF(__pyx_t_4); __pyx_t_4 = 0;
-          if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 200, __pyx_L1_error)
+          if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 209, __pyx_L1_error)
           __Pyx_GOTREF(__pyx_t_2);
           __Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
         }
-        if (!(likely(PyTuple_CheckExact(__pyx_t_2))||((__pyx_t_2) == Py_None) || __Pyx_RaiseUnexpectedTypeError("tuple", __pyx_t_2))) __PYX_ERR(0, 200, __pyx_L1_error)
+        if (!(likely(PyTuple_CheckExact(__pyx_t_2))||((__pyx_t_2) == Py_None) || __Pyx_RaiseUnexpectedTypeError("tuple", __pyx_t_2))) __PYX_ERR(0, 209, __pyx_L1_error)
         __pyx_r = ((PyObject*)__pyx_t_2);
         __pyx_t_2 = 0;
         __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
@@ -25425,25 +25664,25 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
     #endif
   }
 
-  /* "mettagrid/mettagrid.pyx":201
+  /* "mettagrid/mettagrid.pyx":210
  * 
  *     cpdef tuple[cnp.ndarray, cnp.ndarray, cnp.ndarray, cnp.ndarray, dict] step(self, cnp.ndarray actions):
  *         (obs, rewards, terms, truncs, infos) = super(MettaGrid, self).step(actions)             # <<<<<<<<<<<<<<
  * 
  *         self._group_rewards[:] = 0
  */
-  __pyx_t_2 = PyTuple_New(2); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 201, __pyx_L1_error)
+  __pyx_t_2 = PyTuple_New(2); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 210, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_2);
   __Pyx_INCREF((PyObject *)__pyx_ptype_9mettagrid_11mettagrid_c_MettaGrid);
   __Pyx_GIVEREF((PyObject *)__pyx_ptype_9mettagrid_11mettagrid_c_MettaGrid);
-  if (__Pyx_PyTuple_SET_ITEM(__pyx_t_2, 0, ((PyObject *)__pyx_ptype_9mettagrid_11mettagrid_c_MettaGrid))) __PYX_ERR(0, 201, __pyx_L1_error);
+  if (__Pyx_PyTuple_SET_ITEM(__pyx_t_2, 0, ((PyObject *)__pyx_ptype_9mettagrid_11mettagrid_c_MettaGrid))) __PYX_ERR(0, 210, __pyx_L1_error);
   __Pyx_INCREF((PyObject *)__pyx_v_self);
   __Pyx_GIVEREF((PyObject *)__pyx_v_self);
-  if (__Pyx_PyTuple_SET_ITEM(__pyx_t_2, 1, ((PyObject *)__pyx_v_self))) __PYX_ERR(0, 201, __pyx_L1_error);
-  __pyx_t_3 = __Pyx_PyObject_Call(__pyx_builtin_super, __pyx_t_2, NULL); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 201, __pyx_L1_error)
+  if (__Pyx_PyTuple_SET_ITEM(__pyx_t_2, 1, ((PyObject *)__pyx_v_self))) __PYX_ERR(0, 210, __pyx_L1_error);
+  __pyx_t_3 = __Pyx_PyObject_Call(__pyx_builtin_super, __pyx_t_2, NULL); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 210, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_3);
   __Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
-  __pyx_t_2 = __Pyx_PyObject_GetAttrStr(__pyx_t_3, __pyx_n_s_step); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 201, __pyx_L1_error)
+  __pyx_t_2 = __Pyx_PyObject_GetAttrStr(__pyx_t_3, __pyx_n_s_step); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 210, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_2);
   __Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
   __pyx_t_3 = NULL;
@@ -25464,7 +25703,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
     PyObject *__pyx_callargs[2] = {__pyx_t_3, ((PyObject *)__pyx_v_actions)};
     __pyx_t_1 = __Pyx_PyObject_FastCall(__pyx_t_2, __pyx_callargs+1-__pyx_t_5, 1+__pyx_t_5);
     __Pyx_XDECREF(__pyx_t_3); __pyx_t_3 = 0;
-    if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 201, __pyx_L1_error)
+    if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 210, __pyx_L1_error)
     __Pyx_GOTREF(__pyx_t_1);
     __Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
   }
@@ -25474,7 +25713,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
     if (unlikely(size != 5)) {
       if (size > 5) __Pyx_RaiseTooManyValuesError(5);
       else if (size >= 0) __Pyx_RaiseNeedMoreValuesError(size);
-      __PYX_ERR(0, 201, __pyx_L1_error)
+      __PYX_ERR(0, 210, __pyx_L1_error)
     }
     #if CYTHON_ASSUME_SAFE_MACROS && !CYTHON_AVOID_BORROWED_REFS
     if (likely(PyTuple_CheckExact(sequence))) {
@@ -25500,7 +25739,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
       Py_ssize_t i;
       PyObject** temps[5] = {&__pyx_t_2,&__pyx_t_3,&__pyx_t_4,&__pyx_t_6,&__pyx_t_7};
       for (i=0; i < 5; i++) {
-        PyObject* item = PySequence_ITEM(sequence, i); if (unlikely(!item)) __PYX_ERR(0, 201, __pyx_L1_error)
+        PyObject* item = PySequence_ITEM(sequence, i); if (unlikely(!item)) __PYX_ERR(0, 210, __pyx_L1_error)
         __Pyx_GOTREF(item);
         *(temps[i]) = item;
       }
@@ -25510,7 +25749,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
   } else {
     Py_ssize_t index = -1;
     PyObject** temps[5] = {&__pyx_t_2,&__pyx_t_3,&__pyx_t_4,&__pyx_t_6,&__pyx_t_7};
-    __pyx_t_8 = PyObject_GetIter(__pyx_t_1); if (unlikely(!__pyx_t_8)) __PYX_ERR(0, 201, __pyx_L1_error)
+    __pyx_t_8 = PyObject_GetIter(__pyx_t_1); if (unlikely(!__pyx_t_8)) __PYX_ERR(0, 210, __pyx_L1_error)
     __Pyx_GOTREF(__pyx_t_8);
     __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
     __pyx_t_9 = __Pyx_PyObject_GetIterNextFunc(__pyx_t_8);
@@ -25519,7 +25758,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
       __Pyx_GOTREF(item);
       *(temps[index]) = item;
     }
-    if (__Pyx_IternextUnpackEndCheck(__pyx_t_9(__pyx_t_8), 5) < 0) __PYX_ERR(0, 201, __pyx_L1_error)
+    if (__Pyx_IternextUnpackEndCheck(__pyx_t_9(__pyx_t_8), 5) < 0) __PYX_ERR(0, 210, __pyx_L1_error)
     __pyx_t_9 = NULL;
     __Pyx_DECREF(__pyx_t_8); __pyx_t_8 = 0;
     goto __pyx_L4_unpacking_done;
@@ -25527,7 +25766,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
     __Pyx_DECREF(__pyx_t_8); __pyx_t_8 = 0;
     __pyx_t_9 = NULL;
     if (__Pyx_IterFinish() == 0) __Pyx_RaiseNeedMoreValuesError(index);
-    __PYX_ERR(0, 201, __pyx_L1_error)
+    __PYX_ERR(0, 210, __pyx_L1_error)
     __pyx_L4_unpacking_done:;
   }
   __pyx_v_obs = __pyx_t_2;
@@ -25541,7 +25780,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
   __pyx_v_infos = __pyx_t_7;
   __pyx_t_7 = 0;
 
-  /* "mettagrid/mettagrid.pyx":203
+  /* "mettagrid/mettagrid.pyx":212
  *         (obs, rewards, terms, truncs, infos) = super(MettaGrid, self).step(actions)
  * 
  *         self._group_rewards[:] = 0             # <<<<<<<<<<<<<<
@@ -25563,7 +25802,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
       }
   }
 
-  /* "mettagrid/mettagrid.pyx":207
+  /* "mettagrid/mettagrid.pyx":216
  *         cdef unsigned int group_id
  *         cdef float group_reward
  *         cdef bint share_rewards = False             # <<<<<<<<<<<<<<
@@ -25572,7 +25811,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
  */
   __pyx_v_share_rewards = 0;
 
-  /* "mettagrid/mettagrid.pyx":209
+  /* "mettagrid/mettagrid.pyx":218
  *         cdef bint share_rewards = False
  * 
  *         for agent_idx in range(self._agents.size()):             # <<<<<<<<<<<<<<
@@ -25584,20 +25823,20 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
   for (__pyx_t_12 = 0; __pyx_t_12 < __pyx_t_11; __pyx_t_12+=1) {
     __pyx_v_agent_idx = __pyx_t_12;
 
-    /* "mettagrid/mettagrid.pyx":210
+    /* "mettagrid/mettagrid.pyx":219
  * 
  *         for agent_idx in range(self._agents.size()):
  *             if rewards[agent_idx] != 0:             # <<<<<<<<<<<<<<
  *                 share_rewards = True
  *                 agent = <Agent*>self._agents[agent_idx]
  */
-    __pyx_t_1 = __Pyx_GetItemInt(__pyx_v_rewards, __pyx_v_agent_idx, std::vector<GridObject *> ::size_type, 0, __Pyx_PyInt_FromSize_t, 0, 0, 0); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 210, __pyx_L1_error)
+    __pyx_t_1 = __Pyx_GetItemInt(__pyx_v_rewards, __pyx_v_agent_idx, std::vector<GridObject *> ::size_type, 0, __Pyx_PyInt_FromSize_t, 0, 0, 0); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 219, __pyx_L1_error)
     __Pyx_GOTREF(__pyx_t_1);
-    __pyx_t_13 = (__Pyx_PyInt_BoolNeObjC(__pyx_t_1, __pyx_int_0, 0, 0)); if (unlikely((__pyx_t_13 < 0))) __PYX_ERR(0, 210, __pyx_L1_error)
+    __pyx_t_13 = (__Pyx_PyInt_BoolNeObjC(__pyx_t_1, __pyx_int_0, 0, 0)); if (unlikely((__pyx_t_13 < 0))) __PYX_ERR(0, 219, __pyx_L1_error)
     __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
     if (__pyx_t_13) {
 
-      /* "mettagrid/mettagrid.pyx":211
+      /* "mettagrid/mettagrid.pyx":220
  *         for agent_idx in range(self._agents.size()):
  *             if rewards[agent_idx] != 0:
  *                 share_rewards = True             # <<<<<<<<<<<<<<
@@ -25606,7 +25845,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
  */
       __pyx_v_share_rewards = 1;
 
-      /* "mettagrid/mettagrid.pyx":212
+      /* "mettagrid/mettagrid.pyx":221
  *             if rewards[agent_idx] != 0:
  *                 share_rewards = True
  *                 agent = <Agent*>self._agents[agent_idx]             # <<<<<<<<<<<<<<
@@ -25615,7 +25854,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
  */
       __pyx_v_agent = ((Agent *)(__pyx_v_self->__pyx_base._agents[__pyx_v_agent_idx]));
 
-      /* "mettagrid/mettagrid.pyx":213
+      /* "mettagrid/mettagrid.pyx":222
  *                 share_rewards = True
  *                 agent = <Agent*>self._agents[agent_idx]
  *                 group_id = agent.group             # <<<<<<<<<<<<<<
@@ -25625,26 +25864,26 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
       __pyx_t_14 = __pyx_v_agent->group;
       __pyx_v_group_id = __pyx_t_14;
 
-      /* "mettagrid/mettagrid.pyx":214
+      /* "mettagrid/mettagrid.pyx":223
  *                 agent = <Agent*>self._agents[agent_idx]
  *                 group_id = agent.group
  *                 group_reward = rewards[agent_idx] * self._group_reward_pct[group_id]             # <<<<<<<<<<<<<<
  *                 rewards[agent_idx] -= group_reward
  *                 self._group_rewards[group_id] += group_reward / self._group_sizes[group_id]
  */
-      __pyx_t_1 = __Pyx_GetItemInt(__pyx_v_rewards, __pyx_v_agent_idx, std::vector<GridObject *> ::size_type, 0, __Pyx_PyInt_FromSize_t, 0, 0, 0); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 214, __pyx_L1_error)
+      __pyx_t_1 = __Pyx_GetItemInt(__pyx_v_rewards, __pyx_v_agent_idx, std::vector<GridObject *> ::size_type, 0, __Pyx_PyInt_FromSize_t, 0, 0, 0); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 223, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_1);
-      __pyx_t_7 = PyFloat_FromDouble((__pyx_v_self->_group_reward_pct[__pyx_v_group_id])); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 214, __pyx_L1_error)
+      __pyx_t_7 = PyFloat_FromDouble((__pyx_v_self->_group_reward_pct[__pyx_v_group_id])); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 223, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_7);
-      __pyx_t_6 = PyNumber_Multiply(__pyx_t_1, __pyx_t_7); if (unlikely(!__pyx_t_6)) __PYX_ERR(0, 214, __pyx_L1_error)
+      __pyx_t_6 = PyNumber_Multiply(__pyx_t_1, __pyx_t_7); if (unlikely(!__pyx_t_6)) __PYX_ERR(0, 223, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_6);
       __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
       __Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
-      __pyx_t_15 = __pyx_PyFloat_AsFloat(__pyx_t_6); if (unlikely((__pyx_t_15 == (float)-1) && PyErr_Occurred())) __PYX_ERR(0, 214, __pyx_L1_error)
+      __pyx_t_15 = __pyx_PyFloat_AsFloat(__pyx_t_6); if (unlikely((__pyx_t_15 == (float)-1) && PyErr_Occurred())) __PYX_ERR(0, 223, __pyx_L1_error)
       __Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
       __pyx_v_group_reward = __pyx_t_15;
 
-      /* "mettagrid/mettagrid.pyx":215
+      /* "mettagrid/mettagrid.pyx":224
  *                 group_id = agent.group
  *                 group_reward = rewards[agent_idx] * self._group_reward_pct[group_id]
  *                 rewards[agent_idx] -= group_reward             # <<<<<<<<<<<<<<
@@ -25652,18 +25891,18 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
  * 
  */
       __pyx_t_16 = __pyx_v_agent_idx;
-      __pyx_t_6 = __Pyx_GetItemInt(__pyx_v_rewards, __pyx_t_16, std::vector<GridObject *> ::size_type, 0, __Pyx_PyInt_FromSize_t, 0, 0, 0); if (unlikely(!__pyx_t_6)) __PYX_ERR(0, 215, __pyx_L1_error)
+      __pyx_t_6 = __Pyx_GetItemInt(__pyx_v_rewards, __pyx_t_16, std::vector<GridObject *> ::size_type, 0, __Pyx_PyInt_FromSize_t, 0, 0, 0); if (unlikely(!__pyx_t_6)) __PYX_ERR(0, 224, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_6);
-      __pyx_t_7 = PyFloat_FromDouble(__pyx_v_group_reward); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 215, __pyx_L1_error)
+      __pyx_t_7 = PyFloat_FromDouble(__pyx_v_group_reward); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 224, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_7);
-      __pyx_t_1 = PyNumber_InPlaceSubtract(__pyx_t_6, __pyx_t_7); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 215, __pyx_L1_error)
+      __pyx_t_1 = PyNumber_InPlaceSubtract(__pyx_t_6, __pyx_t_7); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 224, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_1);
       __Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
       __Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
-      if (unlikely((__Pyx_SetItemInt(__pyx_v_rewards, __pyx_t_16, __pyx_t_1, std::vector<GridObject *> ::size_type, 0, __Pyx_PyInt_FromSize_t, 0, 0, 0) < 0))) __PYX_ERR(0, 215, __pyx_L1_error)
+      if (unlikely((__Pyx_SetItemInt(__pyx_v_rewards, __pyx_t_16, __pyx_t_1, std::vector<GridObject *> ::size_type, 0, __Pyx_PyInt_FromSize_t, 0, 0, 0) < 0))) __PYX_ERR(0, 224, __pyx_L1_error)
       __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
 
-      /* "mettagrid/mettagrid.pyx":216
+      /* "mettagrid/mettagrid.pyx":225
  *                 group_reward = rewards[agent_idx] * self._group_reward_pct[group_id]
  *                 rewards[agent_idx] -= group_reward
  *                 self._group_rewards[group_id] += group_reward / self._group_sizes[group_id]             # <<<<<<<<<<<<<<
@@ -25673,12 +25912,12 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
       __pyx_t_5 = (__pyx_v_self->_group_sizes[__pyx_v_group_id]);
       if (unlikely(__pyx_t_5 == 0)) {
         PyErr_SetString(PyExc_ZeroDivisionError, "float division");
-        __PYX_ERR(0, 216, __pyx_L1_error)
+        __PYX_ERR(0, 225, __pyx_L1_error)
       }
       __pyx_t_17 = __pyx_v_group_id;
       *((double *) ( /* dim=0 */ (__pyx_v_self->_group_rewards.data + __pyx_t_17 * __pyx_v_self->_group_rewards.strides[0]) )) += (__pyx_v_group_reward / ((float)__pyx_t_5));
 
-      /* "mettagrid/mettagrid.pyx":210
+      /* "mettagrid/mettagrid.pyx":219
  * 
  *         for agent_idx in range(self._agents.size()):
  *             if rewards[agent_idx] != 0:             # <<<<<<<<<<<<<<
@@ -25688,7 +25927,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
     }
   }
 
-  /* "mettagrid/mettagrid.pyx":218
+  /* "mettagrid/mettagrid.pyx":227
  *                 self._group_rewards[group_id] += group_reward / self._group_sizes[group_id]
  * 
  *         if share_rewards:             # <<<<<<<<<<<<<<
@@ -25697,7 +25936,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
  */
   if (__pyx_v_share_rewards) {
 
-    /* "mettagrid/mettagrid.pyx":219
+    /* "mettagrid/mettagrid.pyx":228
  * 
  *         if share_rewards:
  *             for agent_idx in range(self._agents.size()):             # <<<<<<<<<<<<<<
@@ -25709,7 +25948,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
     for (__pyx_t_12 = 0; __pyx_t_12 < __pyx_t_11; __pyx_t_12+=1) {
       __pyx_v_agent_idx = __pyx_t_12;
 
-      /* "mettagrid/mettagrid.pyx":220
+      /* "mettagrid/mettagrid.pyx":229
  *         if share_rewards:
  *             for agent_idx in range(self._agents.size()):
  *                 agent = <Agent*>self._agents[agent_idx]             # <<<<<<<<<<<<<<
@@ -25718,7 +25957,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
  */
       __pyx_v_agent = ((Agent *)(__pyx_v_self->__pyx_base._agents[__pyx_v_agent_idx]));
 
-      /* "mettagrid/mettagrid.pyx":221
+      /* "mettagrid/mettagrid.pyx":230
  *             for agent_idx in range(self._agents.size()):
  *                 agent = <Agent*>self._agents[agent_idx]
  *                 group_id = agent.group             # <<<<<<<<<<<<<<
@@ -25728,7 +25967,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
       __pyx_t_14 = __pyx_v_agent->group;
       __pyx_v_group_id = __pyx_t_14;
 
-      /* "mettagrid/mettagrid.pyx":222
+      /* "mettagrid/mettagrid.pyx":231
  *                 agent = <Agent*>self._agents[agent_idx]
  *                 group_id = agent.group
  *                 group_reward = self._group_rewards[group_id]             # <<<<<<<<<<<<<<
@@ -25738,7 +25977,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
       __pyx_t_17 = __pyx_v_group_id;
       __pyx_v_group_reward = (*((double *) ( /* dim=0 */ (__pyx_v_self->_group_rewards.data + __pyx_t_17 * __pyx_v_self->_group_rewards.strides[0]) )));
 
-      /* "mettagrid/mettagrid.pyx":223
+      /* "mettagrid/mettagrid.pyx":232
  *                 group_id = agent.group
  *                 group_reward = self._group_rewards[group_id]
  *                 rewards[agent_idx] += group_reward             # <<<<<<<<<<<<<<
@@ -25746,19 +25985,19 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
  * 
  */
       __pyx_t_16 = __pyx_v_agent_idx;
-      __pyx_t_1 = __Pyx_GetItemInt(__pyx_v_rewards, __pyx_t_16, std::vector<GridObject *> ::size_type, 0, __Pyx_PyInt_FromSize_t, 0, 0, 0); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 223, __pyx_L1_error)
+      __pyx_t_1 = __Pyx_GetItemInt(__pyx_v_rewards, __pyx_t_16, std::vector<GridObject *> ::size_type, 0, __Pyx_PyInt_FromSize_t, 0, 0, 0); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 232, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_1);
-      __pyx_t_7 = PyFloat_FromDouble(__pyx_v_group_reward); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 223, __pyx_L1_error)
+      __pyx_t_7 = PyFloat_FromDouble(__pyx_v_group_reward); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 232, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_7);
-      __pyx_t_6 = PyNumber_InPlaceAdd(__pyx_t_1, __pyx_t_7); if (unlikely(!__pyx_t_6)) __PYX_ERR(0, 223, __pyx_L1_error)
+      __pyx_t_6 = PyNumber_InPlaceAdd(__pyx_t_1, __pyx_t_7); if (unlikely(!__pyx_t_6)) __PYX_ERR(0, 232, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_6);
       __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
       __Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
-      if (unlikely((__Pyx_SetItemInt(__pyx_v_rewards, __pyx_t_16, __pyx_t_6, std::vector<GridObject *> ::size_type, 0, __Pyx_PyInt_FromSize_t, 0, 0, 0) < 0))) __PYX_ERR(0, 223, __pyx_L1_error)
+      if (unlikely((__Pyx_SetItemInt(__pyx_v_rewards, __pyx_t_16, __pyx_t_6, std::vector<GridObject *> ::size_type, 0, __Pyx_PyInt_FromSize_t, 0, 0, 0) < 0))) __PYX_ERR(0, 232, __pyx_L1_error)
       __Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
     }
 
-    /* "mettagrid/mettagrid.pyx":218
+    /* "mettagrid/mettagrid.pyx":227
  *                 self._group_rewards[group_id] += group_reward / self._group_sizes[group_id]
  * 
  *         if share_rewards:             # <<<<<<<<<<<<<<
@@ -25767,7 +26006,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
  */
   }
 
-  /* "mettagrid/mettagrid.pyx":224
+  /* "mettagrid/mettagrid.pyx":233
  *                 group_reward = self._group_rewards[group_id]
  *                 rewards[agent_idx] += group_reward
  *         return (obs, rewards, terms, truncs, infos)             # <<<<<<<<<<<<<<
@@ -25775,28 +26014,28 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(struct __pyx_o
  *     cpdef dict get_episode_stats(self):
  */
   __Pyx_XDECREF(__pyx_r);
-  __pyx_t_6 = PyTuple_New(5); if (unlikely(!__pyx_t_6)) __PYX_ERR(0, 224, __pyx_L1_error)
+  __pyx_t_6 = PyTuple_New(5); if (unlikely(!__pyx_t_6)) __PYX_ERR(0, 233, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_6);
   __Pyx_INCREF(__pyx_v_obs);
   __Pyx_GIVEREF(__pyx_v_obs);
-  if (__Pyx_PyTuple_SET_ITEM(__pyx_t_6, 0, __pyx_v_obs)) __PYX_ERR(0, 224, __pyx_L1_error);
+  if (__Pyx_PyTuple_SET_ITEM(__pyx_t_6, 0, __pyx_v_obs)) __PYX_ERR(0, 233, __pyx_L1_error);
   __Pyx_INCREF(__pyx_v_rewards);
   __Pyx_GIVEREF(__pyx_v_rewards);
-  if (__Pyx_PyTuple_SET_ITEM(__pyx_t_6, 1, __pyx_v_rewards)) __PYX_ERR(0, 224, __pyx_L1_error);
+  if (__Pyx_PyTuple_SET_ITEM(__pyx_t_6, 1, __pyx_v_rewards)) __PYX_ERR(0, 233, __pyx_L1_error);
   __Pyx_INCREF(__pyx_v_terms);
   __Pyx_GIVEREF(__pyx_v_terms);
-  if (__Pyx_PyTuple_SET_ITEM(__pyx_t_6, 2, __pyx_v_terms)) __PYX_ERR(0, 224, __pyx_L1_error);
+  if (__Pyx_PyTuple_SET_ITEM(__pyx_t_6, 2, __pyx_v_terms)) __PYX_ERR(0, 233, __pyx_L1_error);
   __Pyx_INCREF(__pyx_v_truncs);
   __Pyx_GIVEREF(__pyx_v_truncs);
-  if (__Pyx_PyTuple_SET_ITEM(__pyx_t_6, 3, __pyx_v_truncs)) __PYX_ERR(0, 224, __pyx_L1_error);
+  if (__Pyx_PyTuple_SET_ITEM(__pyx_t_6, 3, __pyx_v_truncs)) __PYX_ERR(0, 233, __pyx_L1_error);
   __Pyx_INCREF(__pyx_v_infos);
   __Pyx_GIVEREF(__pyx_v_infos);
-  if (__Pyx_PyTuple_SET_ITEM(__pyx_t_6, 4, __pyx_v_infos)) __PYX_ERR(0, 224, __pyx_L1_error);
+  if (__Pyx_PyTuple_SET_ITEM(__pyx_t_6, 4, __pyx_v_infos)) __PYX_ERR(0, 233, __pyx_L1_error);
   __pyx_r = ((PyObject*)__pyx_t_6);
   __pyx_t_6 = 0;
   goto __pyx_L0;
 
-  /* "mettagrid/mettagrid.pyx":200
+  /* "mettagrid/mettagrid.pyx":209
  *         return objects
  * 
  *     cpdef tuple[cnp.ndarray, cnp.ndarray, cnp.ndarray, cnp.ndarray, dict] step(self, cnp.ndarray actions):             # <<<<<<<<<<<<<<
@@ -25879,12 +26118,12 @@ PyObject *__pyx_args, PyObject *__pyx_kwds
           (void)__Pyx_Arg_NewRef_FASTCALL(values[0]);
           kw_args--;
         }
-        else if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 200, __pyx_L3_error)
+        else if (unlikely(PyErr_Occurred())) __PYX_ERR(0, 209, __pyx_L3_error)
         else goto __pyx_L5_argtuple_error;
       }
       if (unlikely(kw_args > 0)) {
         const Py_ssize_t kwd_pos_args = __pyx_nargs;
-        if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_kwvalues, __pyx_pyargnames, 0, values + 0, kwd_pos_args, "step") < 0)) __PYX_ERR(0, 200, __pyx_L3_error)
+        if (unlikely(__Pyx_ParseOptionalKeywords(__pyx_kwds, __pyx_kwvalues, __pyx_pyargnames, 0, values + 0, kwd_pos_args, "step") < 0)) __PYX_ERR(0, 209, __pyx_L3_error)
       }
     } else if (unlikely(__pyx_nargs != 1)) {
       goto __pyx_L5_argtuple_error;
@@ -25895,7 +26134,7 @@ PyObject *__pyx_args, PyObject *__pyx_kwds
   }
   goto __pyx_L6_skip;
   __pyx_L5_argtuple_error:;
-  __Pyx_RaiseArgtupleInvalid("step", 1, 1, 1, __pyx_nargs); __PYX_ERR(0, 200, __pyx_L3_error)
+  __Pyx_RaiseArgtupleInvalid("step", 1, 1, 1, __pyx_nargs); __PYX_ERR(0, 209, __pyx_L3_error)
   __pyx_L6_skip:;
   goto __pyx_L4_argument_unpacking_done;
   __pyx_L3_error:;
@@ -25909,7 +26148,7 @@ PyObject *__pyx_args, PyObject *__pyx_kwds
   __Pyx_RefNannyFinishContext();
   return NULL;
   __pyx_L4_argument_unpacking_done:;
-  if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_actions), __pyx_ptype_5numpy_ndarray, 1, "actions", 0))) __PYX_ERR(0, 200, __pyx_L1_error)
+  if (unlikely(!__Pyx_ArgTypeTest(((PyObject *)__pyx_v_actions), __pyx_ptype_5numpy_ndarray, 1, "actions", 0))) __PYX_ERR(0, 209, __pyx_L1_error)
   __pyx_r = __pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid_8step(((struct __pyx_obj_9mettagrid_11mettagrid_c_MettaGrid *)__pyx_v_self), __pyx_v_actions);
 
   /* function exit code */
@@ -25936,7 +26175,7 @@ static PyObject *__pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid_8step(struct __pyx
   int __pyx_clineno = 0;
   __Pyx_RefNannySetupContext("step", 1);
   __Pyx_XDECREF(__pyx_r);
-  __pyx_t_1 = __pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(__pyx_v_self, __pyx_v_actions, 1); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 200, __pyx_L1_error)
+  __pyx_t_1 = __pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_step(__pyx_v_self, __pyx_v_actions, 1); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 209, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_1);
   __pyx_r = __pyx_t_1;
   __pyx_t_1 = 0;
@@ -25953,7 +26192,7 @@ static PyObject *__pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid_8step(struct __pyx
   return __pyx_r;
 }
 
-/* "mettagrid/mettagrid.pyx":226
+/* "mettagrid/mettagrid.pyx":235
  *         return (obs, rewards, terms, truncs, infos)
  * 
  *     cpdef dict get_episode_stats(self):             # <<<<<<<<<<<<<<
@@ -25992,7 +26231,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_get_episode_stats(s
     if (unlikely(!__Pyx_object_dict_version_matches(((PyObject *)__pyx_v_self), __pyx_tp_dict_version, __pyx_obj_dict_version))) {
       PY_UINT64_T __pyx_typedict_guard = __Pyx_get_tp_dict_version(((PyObject *)__pyx_v_self));
       #endif
-      __pyx_t_1 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_get_episode_stats); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 226, __pyx_L1_error)
+      __pyx_t_1 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_get_episode_stats); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 235, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_1);
       if (!__Pyx_IsSameCFunction(__pyx_t_1, (void*) __pyx_pw_9mettagrid_11mettagrid_c_9MettaGrid_11get_episode_stats)) {
         __Pyx_XDECREF(__pyx_r);
@@ -26015,11 +26254,11 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_get_episode_stats(s
           PyObject *__pyx_callargs[2] = {__pyx_t_4, NULL};
           __pyx_t_2 = __Pyx_PyObject_FastCall(__pyx_t_3, __pyx_callargs+1-__pyx_t_5, 0+__pyx_t_5);
           __Pyx_XDECREF(__pyx_t_4); __pyx_t_4 = 0;
-          if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 226, __pyx_L1_error)
+          if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 235, __pyx_L1_error)
           __Pyx_GOTREF(__pyx_t_2);
           __Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
         }
-        if (!(likely(PyDict_CheckExact(__pyx_t_2))||((__pyx_t_2) == Py_None) || __Pyx_RaiseUnexpectedTypeError("dict", __pyx_t_2))) __PYX_ERR(0, 226, __pyx_L1_error)
+        if (!(likely(PyDict_CheckExact(__pyx_t_2))||((__pyx_t_2) == Py_None) || __Pyx_RaiseUnexpectedTypeError("dict", __pyx_t_2))) __PYX_ERR(0, 235, __pyx_L1_error)
         __pyx_r = ((PyObject*)__pyx_t_2);
         __pyx_t_2 = 0;
         __Pyx_DECREF(__pyx_t_1); __pyx_t_1 = 0;
@@ -26038,7 +26277,7 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_get_episode_stats(s
     #endif
   }
 
-  /* "mettagrid/mettagrid.pyx":227
+  /* "mettagrid/mettagrid.pyx":236
  * 
  *     cpdef dict get_episode_stats(self):
  *         return {             # <<<<<<<<<<<<<<
@@ -26047,28 +26286,28 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_get_episode_stats(s
  */
   __Pyx_XDECREF(__pyx_r);
 
-  /* "mettagrid/mettagrid.pyx":228
+  /* "mettagrid/mettagrid.pyx":237
  *     cpdef dict get_episode_stats(self):
  *         return {
  *             "game": self._stats.stats(),             # <<<<<<<<<<<<<<
  *             "agent": [ (<Agent*>agent).stats.stats() for agent in self._agents ]
  *         }
  */
-  __pyx_t_1 = __Pyx_PyDict_NewPresized(2); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 228, __pyx_L1_error)
+  __pyx_t_1 = __Pyx_PyDict_NewPresized(2); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 237, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_1);
-  __pyx_t_2 = __pyx_convert_map_to_py_std_3a__3a_string____int(__pyx_v_self->__pyx_base._stats.stats()); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 228, __pyx_L1_error)
+  __pyx_t_2 = __pyx_convert_map_to_py_std_3a__3a_string____int(__pyx_v_self->__pyx_base._stats.stats()); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 237, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_2);
-  if (PyDict_SetItem(__pyx_t_1, __pyx_n_u_game, __pyx_t_2) < 0) __PYX_ERR(0, 228, __pyx_L1_error)
+  if (PyDict_SetItem(__pyx_t_1, __pyx_n_u_game, __pyx_t_2) < 0) __PYX_ERR(0, 237, __pyx_L1_error)
   __Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
   { /* enter inner scope */
 
-    /* "mettagrid/mettagrid.pyx":229
+    /* "mettagrid/mettagrid.pyx":238
  *         return {
  *             "game": self._stats.stats(),
  *             "agent": [ (<Agent*>agent).stats.stats() for agent in self._agents ]             # <<<<<<<<<<<<<<
  *         }
  */
-    __pyx_t_2 = PyList_New(0); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 229, __pyx_L1_error)
+    __pyx_t_2 = PyList_New(0); if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 238, __pyx_L1_error)
     __Pyx_GOTREF(__pyx_t_2);
     __pyx_t_6 = __pyx_v_self->__pyx_base._agents.begin();
     for (;;) {
@@ -26076,19 +26315,19 @@ static PyObject *__pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_get_episode_stats(s
       __pyx_t_7 = *__pyx_t_6;
       ++__pyx_t_6;
       __pyx_8genexpr2__pyx_v_agent = __pyx_t_7;
-      __pyx_t_3 = __pyx_convert_map_to_py_std_3a__3a_string____int(((Agent *)__pyx_8genexpr2__pyx_v_agent)->stats.stats()); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 229, __pyx_L1_error)
+      __pyx_t_3 = __pyx_convert_map_to_py_std_3a__3a_string____int(((Agent *)__pyx_8genexpr2__pyx_v_agent)->stats.stats()); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 238, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_3);
-      if (unlikely(__Pyx_ListComp_Append(__pyx_t_2, (PyObject*)__pyx_t_3))) __PYX_ERR(0, 229, __pyx_L1_error)
+      if (unlikely(__Pyx_ListComp_Append(__pyx_t_2, (PyObject*)__pyx_t_3))) __PYX_ERR(0, 238, __pyx_L1_error)
       __Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
     }
   } /* exit inner scope */
-  if (PyDict_SetItem(__pyx_t_1, __pyx_n_u_agent_2, __pyx_t_2) < 0) __PYX_ERR(0, 228, __pyx_L1_error)
+  if (PyDict_SetItem(__pyx_t_1, __pyx_n_u_agent_2, __pyx_t_2) < 0) __PYX_ERR(0, 237, __pyx_L1_error)
   __Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
   __pyx_r = ((PyObject*)__pyx_t_1);
   __pyx_t_1 = 0;
   goto __pyx_L0;
 
-  /* "mettagrid/mettagrid.pyx":226
+  /* "mettagrid/mettagrid.pyx":235
  *         return (obs, rewards, terms, truncs, infos)
  * 
  *     cpdef dict get_episode_stats(self):             # <<<<<<<<<<<<<<
@@ -26160,7 +26399,7 @@ static PyObject *__pyx_pf_9mettagrid_11mettagrid_c_9MettaGrid_10get_episode_stat
   int __pyx_clineno = 0;
   __Pyx_RefNannySetupContext("get_episode_stats", 1);
   __Pyx_XDECREF(__pyx_r);
-  __pyx_t_1 = __pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_get_episode_stats(__pyx_v_self, 1); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 226, __pyx_L1_error)
+  __pyx_t_1 = __pyx_f_9mettagrid_11mettagrid_c_9MettaGrid_get_episode_stats(__pyx_v_self, 1); if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 235, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_1);
   __pyx_r = __pyx_t_1;
   __pyx_t_1 = 0;
@@ -27618,6 +27857,8 @@ static int __Pyx_CreateStringTabAndInitStrings(void) {
     {&__pyx_n_s_asyncio_coroutines, __pyx_k_asyncio_coroutines, sizeof(__pyx_k_asyncio_coroutines), 0, 0, 1, 1},
     {&__pyx_n_s_attack, __pyx_k_attack, sizeof(__pyx_k_attack), 0, 0, 1, 1},
     {&__pyx_n_s_base, __pyx_k_base, sizeof(__pyx_k_base), 0, 0, 1, 1},
+    {&__pyx_n_s_block, __pyx_k_block, sizeof(__pyx_k_block), 0, 0, 1, 1},
+    {&__pyx_n_u_block, __pyx_k_block, sizeof(__pyx_k_block), 0, 1, 0, 1},
     {&__pyx_n_s_c, __pyx_k_c, sizeof(__pyx_k_c), 0, 0, 1, 1},
     {&__pyx_n_u_c, __pyx_k_c, sizeof(__pyx_k_c), 0, 1, 0, 1},
     {&__pyx_n_s_change_color, __pyx_k_change_color, sizeof(__pyx_k_change_color), 0, 0, 1, 1},
@@ -27650,8 +27891,8 @@ static int __Pyx_CreateStringTabAndInitStrings(void) {
     {&__pyx_n_s_game, __pyx_k_game, sizeof(__pyx_k_game), 0, 0, 1, 1},
     {&__pyx_n_u_game, __pyx_k_game, sizeof(__pyx_k_game), 0, 1, 0, 1},
     {&__pyx_kp_u_gc, __pyx_k_gc, sizeof(__pyx_k_gc), 0, 1, 0, 0},
-    {&__pyx_n_s_generator, __pyx_k_generator, sizeof(__pyx_k_generator), 0, 0, 1, 1},
     {&__pyx_n_u_generator, __pyx_k_generator, sizeof(__pyx_k_generator), 0, 1, 0, 1},
+    {&__pyx_kp_u_generator_red, __pyx_k_generator_red, sizeof(__pyx_k_generator_red), 0, 1, 0, 0},
     {&__pyx_n_s_get, __pyx_k_get, sizeof(__pyx_k_get), 0, 0, 1, 1},
     {&__pyx_n_s_get_episode_stats, __pyx_k_get_episode_stats, sizeof(__pyx_k_get_episode_stats), 0, 0, 1, 1},
     {&__pyx_n_s_get_items, __pyx_k_get_items, sizeof(__pyx_k_get_items), 0, 0, 1, 1},
@@ -27699,8 +27940,8 @@ static int __Pyx_CreateStringTabAndInitStrings(void) {
     {&__pyx_n_s_mettagrid_actions_swap, __pyx_k_mettagrid_actions_swap, sizeof(__pyx_k_mettagrid_actions_swap), 0, 0, 1, 1},
     {&__pyx_n_s_mettagrid_mettagrid_c, __pyx_k_mettagrid_mettagrid_c, sizeof(__pyx_k_mettagrid_mettagrid_c), 0, 0, 1, 1},
     {&__pyx_kp_s_mettagrid_mettagrid_pyx, __pyx_k_mettagrid_mettagrid_pyx, sizeof(__pyx_k_mettagrid_mettagrid_pyx), 0, 0, 1, 0},
-    {&__pyx_n_s_mine, __pyx_k_mine, sizeof(__pyx_k_mine), 0, 0, 1, 1},
     {&__pyx_n_u_mine, __pyx_k_mine, sizeof(__pyx_k_mine), 0, 1, 0, 1},
+    {&__pyx_kp_u_mine_red, __pyx_k_mine_red, sizeof(__pyx_k_mine_red), 0, 1, 0, 0},
     {&__pyx_n_s_mode, __pyx_k_mode, sizeof(__pyx_k_mode), 0, 0, 1, 1},
     {&__pyx_n_s_move, __pyx_k_move, sizeof(__pyx_k_move), 0, 0, 1, 1},
     {&__pyx_n_s_name, __pyx_k_name, sizeof(__pyx_k_name), 0, 0, 1, 1},
@@ -27786,9 +28027,9 @@ static int __Pyx_CreateStringTabAndInitStrings(void) {
 /* #### Code section: cached_builtins ### */
 static CYTHON_SMALL_CODE int __Pyx_InitCachedBuiltins(void) {
   __pyx_builtin_range = __Pyx_GetBuiltinName(__pyx_n_s_range); if (!__pyx_builtin_range) __PYX_ERR(0, 110, __pyx_L1_error)
-  __pyx_builtin_print = __Pyx_GetBuiltinName(__pyx_n_s_print); if (!__pyx_builtin_print) __PYX_ERR(0, 167, __pyx_L1_error)
-  __pyx_builtin_enumerate = __Pyx_GetBuiltinName(__pyx_n_s_enumerate); if (!__pyx_builtin_enumerate) __PYX_ERR(0, 191, __pyx_L1_error)
-  __pyx_builtin_super = __Pyx_GetBuiltinName(__pyx_n_s_super); if (!__pyx_builtin_super) __PYX_ERR(0, 201, __pyx_L1_error)
+  __pyx_builtin_print = __Pyx_GetBuiltinName(__pyx_n_s_print); if (!__pyx_builtin_print) __PYX_ERR(0, 176, __pyx_L1_error)
+  __pyx_builtin_enumerate = __Pyx_GetBuiltinName(__pyx_n_s_enumerate); if (!__pyx_builtin_enumerate) __PYX_ERR(0, 200, __pyx_L1_error)
+  __pyx_builtin_super = __Pyx_GetBuiltinName(__pyx_n_s_super); if (!__pyx_builtin_super) __PYX_ERR(0, 210, __pyx_L1_error)
   __pyx_builtin_TypeError = __Pyx_GetBuiltinName(__pyx_n_s_TypeError); if (!__pyx_builtin_TypeError) __PYX_ERR(1, 2, __pyx_L1_error)
   __pyx_builtin_MemoryError = __Pyx_GetBuiltinName(__pyx_n_s_MemoryError); if (!__pyx_builtin_MemoryError) __PYX_ERR(1, 68, __pyx_L1_error)
   __pyx_builtin___import__ = __Pyx_GetBuiltinName(__pyx_n_s_import); if (!__pyx_builtin___import__) __PYX_ERR(1, 100, __pyx_L1_error)
@@ -27978,59 +28219,59 @@ static CYTHON_SMALL_CODE int __Pyx_InitCachedConstants(void) {
   __Pyx_GIVEREF(__pyx_tuple__23);
   __pyx_codeobj__24 = (PyObject*)__Pyx_PyCode_New(3, 0, 0, 5, 0, CO_OPTIMIZED|CO_NEWLOCALS, __pyx_empty_bytes, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_tuple__23, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_kp_s_stringsource, __pyx_n_s_pyx_unpickle_Enum, 1, __pyx_empty_bytes); if (unlikely(!__pyx_codeobj__24)) __PYX_ERR(1, 1, __pyx_L1_error)
 
-  /* "mettagrid/mettagrid.pyx":161
+  /* "mettagrid/mettagrid.pyx":170
  * 
  * 
  *     cpdef list[str] grid_features(self):             # <<<<<<<<<<<<<<
  *         return self._grid_features
  * 
  */
-  __pyx_tuple__25 = PyTuple_Pack(1, __pyx_n_s_self); if (unlikely(!__pyx_tuple__25)) __PYX_ERR(0, 161, __pyx_L1_error)
+  __pyx_tuple__25 = PyTuple_Pack(1, __pyx_n_s_self); if (unlikely(!__pyx_tuple__25)) __PYX_ERR(0, 170, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_tuple__25);
   __Pyx_GIVEREF(__pyx_tuple__25);
-  __pyx_codeobj__26 = (PyObject*)__Pyx_PyCode_New(1, 0, 0, 1, 0, CO_OPTIMIZED|CO_NEWLOCALS, __pyx_empty_bytes, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_tuple__25, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_kp_s_mettagrid_mettagrid_pyx, __pyx_n_s_grid_features, 161, __pyx_empty_bytes); if (unlikely(!__pyx_codeobj__26)) __PYX_ERR(0, 161, __pyx_L1_error)
+  __pyx_codeobj__26 = (PyObject*)__Pyx_PyCode_New(1, 0, 0, 1, 0, CO_OPTIMIZED|CO_NEWLOCALS, __pyx_empty_bytes, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_tuple__25, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_kp_s_mettagrid_mettagrid_pyx, __pyx_n_s_grid_features, 170, __pyx_empty_bytes); if (unlikely(!__pyx_codeobj__26)) __PYX_ERR(0, 170, __pyx_L1_error)
 
-  /* "mettagrid/mettagrid.pyx":164
+  /* "mettagrid/mettagrid.pyx":173
  *         return self._grid_features
  * 
  *     def render(self):             # <<<<<<<<<<<<<<
  *         grid = self.render_ascii(["A", "#", "g", "c", "a"])
  *         for r in grid:
  */
-  __pyx_tuple__27 = PyTuple_Pack(3, __pyx_n_s_self, __pyx_n_s_grid, __pyx_n_s_r); if (unlikely(!__pyx_tuple__27)) __PYX_ERR(0, 164, __pyx_L1_error)
+  __pyx_tuple__27 = PyTuple_Pack(3, __pyx_n_s_self, __pyx_n_s_grid, __pyx_n_s_r); if (unlikely(!__pyx_tuple__27)) __PYX_ERR(0, 173, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_tuple__27);
   __Pyx_GIVEREF(__pyx_tuple__27);
-  __pyx_codeobj__28 = (PyObject*)__Pyx_PyCode_New(1, 0, 0, 3, 0, CO_OPTIMIZED|CO_NEWLOCALS, __pyx_empty_bytes, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_tuple__27, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_kp_s_mettagrid_mettagrid_pyx, __pyx_n_s_render, 164, __pyx_empty_bytes); if (unlikely(!__pyx_codeobj__28)) __PYX_ERR(0, 164, __pyx_L1_error)
+  __pyx_codeobj__28 = (PyObject*)__Pyx_PyCode_New(1, 0, 0, 3, 0, CO_OPTIMIZED|CO_NEWLOCALS, __pyx_empty_bytes, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_tuple__27, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_kp_s_mettagrid_mettagrid_pyx, __pyx_n_s_render, 173, __pyx_empty_bytes); if (unlikely(!__pyx_codeobj__28)) __PYX_ERR(0, 173, __pyx_L1_error)
 
-  /* "mettagrid/mettagrid.pyx":169
+  /* "mettagrid/mettagrid.pyx":178
  *                 print("".join(r))
  * 
  *     cpdef grid_objects(self):             # <<<<<<<<<<<<<<
  *         cdef GridObject *obj
  *         cdef ObsType[:] obj_data = np.zeros(len(self.grid_features()), dtype=self._obs_encoder.obs_np_type())
  */
-  __pyx_codeobj__29 = (PyObject*)__Pyx_PyCode_New(1, 0, 0, 1, 0, CO_OPTIMIZED|CO_NEWLOCALS, __pyx_empty_bytes, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_tuple__25, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_kp_s_mettagrid_mettagrid_pyx, __pyx_n_s_grid_objects, 169, __pyx_empty_bytes); if (unlikely(!__pyx_codeobj__29)) __PYX_ERR(0, 169, __pyx_L1_error)
+  __pyx_codeobj__29 = (PyObject*)__Pyx_PyCode_New(1, 0, 0, 1, 0, CO_OPTIMIZED|CO_NEWLOCALS, __pyx_empty_bytes, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_tuple__25, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_kp_s_mettagrid_mettagrid_pyx, __pyx_n_s_grid_objects, 178, __pyx_empty_bytes); if (unlikely(!__pyx_codeobj__29)) __PYX_ERR(0, 178, __pyx_L1_error)
 
-  /* "mettagrid/mettagrid.pyx":200
+  /* "mettagrid/mettagrid.pyx":209
  *         return objects
  * 
  *     cpdef tuple[cnp.ndarray, cnp.ndarray, cnp.ndarray, cnp.ndarray, dict] step(self, cnp.ndarray actions):             # <<<<<<<<<<<<<<
  *         (obs, rewards, terms, truncs, infos) = super(MettaGrid, self).step(actions)
  * 
  */
-  __pyx_tuple__30 = PyTuple_Pack(2, __pyx_n_s_self, __pyx_n_s_actions); if (unlikely(!__pyx_tuple__30)) __PYX_ERR(0, 200, __pyx_L1_error)
+  __pyx_tuple__30 = PyTuple_Pack(2, __pyx_n_s_self, __pyx_n_s_actions); if (unlikely(!__pyx_tuple__30)) __PYX_ERR(0, 209, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_tuple__30);
   __Pyx_GIVEREF(__pyx_tuple__30);
-  __pyx_codeobj__31 = (PyObject*)__Pyx_PyCode_New(2, 0, 0, 2, 0, CO_OPTIMIZED|CO_NEWLOCALS, __pyx_empty_bytes, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_tuple__30, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_kp_s_mettagrid_mettagrid_pyx, __pyx_n_s_step, 200, __pyx_empty_bytes); if (unlikely(!__pyx_codeobj__31)) __PYX_ERR(0, 200, __pyx_L1_error)
+  __pyx_codeobj__31 = (PyObject*)__Pyx_PyCode_New(2, 0, 0, 2, 0, CO_OPTIMIZED|CO_NEWLOCALS, __pyx_empty_bytes, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_tuple__30, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_kp_s_mettagrid_mettagrid_pyx, __pyx_n_s_step, 209, __pyx_empty_bytes); if (unlikely(!__pyx_codeobj__31)) __PYX_ERR(0, 209, __pyx_L1_error)
 
-  /* "mettagrid/mettagrid.pyx":226
+  /* "mettagrid/mettagrid.pyx":235
  *         return (obs, rewards, terms, truncs, infos)
  * 
  *     cpdef dict get_episode_stats(self):             # <<<<<<<<<<<<<<
  *         return {
  *             "game": self._stats.stats(),
  */
-  __pyx_codeobj__32 = (PyObject*)__Pyx_PyCode_New(1, 0, 0, 1, 0, CO_OPTIMIZED|CO_NEWLOCALS, __pyx_empty_bytes, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_tuple__25, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_kp_s_mettagrid_mettagrid_pyx, __pyx_n_s_get_episode_stats, 226, __pyx_empty_bytes); if (unlikely(!__pyx_codeobj__32)) __PYX_ERR(0, 226, __pyx_L1_error)
+  __pyx_codeobj__32 = (PyObject*)__Pyx_PyCode_New(1, 0, 0, 1, 0, CO_OPTIMIZED|CO_NEWLOCALS, __pyx_empty_bytes, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_tuple__25, __pyx_empty_tuple, __pyx_empty_tuple, __pyx_kp_s_mettagrid_mettagrid_pyx, __pyx_n_s_get_episode_stats, 235, __pyx_empty_bytes); if (unlikely(!__pyx_codeobj__32)) __PYX_ERR(0, 235, __pyx_L1_error)
 
   /* "(tree fragment)":1
  * def __reduce_cython__(self):             # <<<<<<<<<<<<<<
@@ -29469,68 +29710,68 @@ if (!__Pyx_RefNanny) {
   __Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
   __Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
 
-  /* "mettagrid/mettagrid.pyx":161
+  /* "mettagrid/mettagrid.pyx":170
  * 
  * 
  *     cpdef list[str] grid_features(self):             # <<<<<<<<<<<<<<
  *         return self._grid_features
  * 
  */
-  __pyx_t_4 = __Pyx_CyFunction_New(&__pyx_mdef_9mettagrid_11mettagrid_c_9MettaGrid_3grid_features, __Pyx_CYFUNCTION_CCLASS, __pyx_n_s_MettaGrid_grid_features, NULL, __pyx_n_s_mettagrid_mettagrid_c, __pyx_d, ((PyObject *)__pyx_codeobj__26)); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 161, __pyx_L1_error)
+  __pyx_t_4 = __Pyx_CyFunction_New(&__pyx_mdef_9mettagrid_11mettagrid_c_9MettaGrid_3grid_features, __Pyx_CYFUNCTION_CCLASS, __pyx_n_s_MettaGrid_grid_features, NULL, __pyx_n_s_mettagrid_mettagrid_c, __pyx_d, ((PyObject *)__pyx_codeobj__26)); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 170, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_4);
-  if (__Pyx_SetItemOnTypeDict((PyObject *)__pyx_ptype_9mettagrid_11mettagrid_c_MettaGrid, __pyx_n_s_grid_features, __pyx_t_4) < 0) __PYX_ERR(0, 161, __pyx_L1_error)
+  if (__Pyx_SetItemOnTypeDict((PyObject *)__pyx_ptype_9mettagrid_11mettagrid_c_MettaGrid, __pyx_n_s_grid_features, __pyx_t_4) < 0) __PYX_ERR(0, 170, __pyx_L1_error)
   __Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
   PyType_Modified(__pyx_ptype_9mettagrid_11mettagrid_c_MettaGrid);
 
-  /* "mettagrid/mettagrid.pyx":164
+  /* "mettagrid/mettagrid.pyx":173
  *         return self._grid_features
  * 
  *     def render(self):             # <<<<<<<<<<<<<<
  *         grid = self.render_ascii(["A", "#", "g", "c", "a"])
  *         for r in grid:
  */
-  __pyx_t_4 = __Pyx_CyFunction_New(&__pyx_mdef_9mettagrid_11mettagrid_c_9MettaGrid_5render, __Pyx_CYFUNCTION_CCLASS, __pyx_n_s_MettaGrid_render, NULL, __pyx_n_s_mettagrid_mettagrid_c, __pyx_d, ((PyObject *)__pyx_codeobj__28)); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 164, __pyx_L1_error)
+  __pyx_t_4 = __Pyx_CyFunction_New(&__pyx_mdef_9mettagrid_11mettagrid_c_9MettaGrid_5render, __Pyx_CYFUNCTION_CCLASS, __pyx_n_s_MettaGrid_render, NULL, __pyx_n_s_mettagrid_mettagrid_c, __pyx_d, ((PyObject *)__pyx_codeobj__28)); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 173, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_4);
-  if (__Pyx_SetItemOnTypeDict((PyObject *)__pyx_ptype_9mettagrid_11mettagrid_c_MettaGrid, __pyx_n_s_render, __pyx_t_4) < 0) __PYX_ERR(0, 164, __pyx_L1_error)
+  if (__Pyx_SetItemOnTypeDict((PyObject *)__pyx_ptype_9mettagrid_11mettagrid_c_MettaGrid, __pyx_n_s_render, __pyx_t_4) < 0) __PYX_ERR(0, 173, __pyx_L1_error)
   __Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
   PyType_Modified(__pyx_ptype_9mettagrid_11mettagrid_c_MettaGrid);
 
-  /* "mettagrid/mettagrid.pyx":169
+  /* "mettagrid/mettagrid.pyx":178
  *                 print("".join(r))
  * 
  *     cpdef grid_objects(self):             # <<<<<<<<<<<<<<
  *         cdef GridObject *obj
  *         cdef ObsType[:] obj_data = np.zeros(len(self.grid_features()), dtype=self._obs_encoder.obs_np_type())
  */
-  __pyx_t_4 = __Pyx_CyFunction_New(&__pyx_mdef_9mettagrid_11mettagrid_c_9MettaGrid_7grid_objects, __Pyx_CYFUNCTION_CCLASS, __pyx_n_s_MettaGrid_grid_objects, NULL, __pyx_n_s_mettagrid_mettagrid_c, __pyx_d, ((PyObject *)__pyx_codeobj__29)); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 169, __pyx_L1_error)
+  __pyx_t_4 = __Pyx_CyFunction_New(&__pyx_mdef_9mettagrid_11mettagrid_c_9MettaGrid_7grid_objects, __Pyx_CYFUNCTION_CCLASS, __pyx_n_s_MettaGrid_grid_objects, NULL, __pyx_n_s_mettagrid_mettagrid_c, __pyx_d, ((PyObject *)__pyx_codeobj__29)); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 178, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_4);
-  if (__Pyx_SetItemOnTypeDict((PyObject *)__pyx_ptype_9mettagrid_11mettagrid_c_MettaGrid, __pyx_n_s_grid_objects, __pyx_t_4) < 0) __PYX_ERR(0, 169, __pyx_L1_error)
+  if (__Pyx_SetItemOnTypeDict((PyObject *)__pyx_ptype_9mettagrid_11mettagrid_c_MettaGrid, __pyx_n_s_grid_objects, __pyx_t_4) < 0) __PYX_ERR(0, 178, __pyx_L1_error)
   __Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
   PyType_Modified(__pyx_ptype_9mettagrid_11mettagrid_c_MettaGrid);
 
-  /* "mettagrid/mettagrid.pyx":200
+  /* "mettagrid/mettagrid.pyx":209
  *         return objects
  * 
  *     cpdef tuple[cnp.ndarray, cnp.ndarray, cnp.ndarray, cnp.ndarray, dict] step(self, cnp.ndarray actions):             # <<<<<<<<<<<<<<
  *         (obs, rewards, terms, truncs, infos) = super(MettaGrid, self).step(actions)
  * 
  */
-  __pyx_t_4 = __Pyx_CyFunction_New(&__pyx_mdef_9mettagrid_11mettagrid_c_9MettaGrid_9step, __Pyx_CYFUNCTION_CCLASS, __pyx_n_s_MettaGrid_step, NULL, __pyx_n_s_mettagrid_mettagrid_c, __pyx_d, ((PyObject *)__pyx_codeobj__31)); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 200, __pyx_L1_error)
+  __pyx_t_4 = __Pyx_CyFunction_New(&__pyx_mdef_9mettagrid_11mettagrid_c_9MettaGrid_9step, __Pyx_CYFUNCTION_CCLASS, __pyx_n_s_MettaGrid_step, NULL, __pyx_n_s_mettagrid_mettagrid_c, __pyx_d, ((PyObject *)__pyx_codeobj__31)); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 209, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_4);
-  if (__Pyx_SetItemOnTypeDict((PyObject *)__pyx_ptype_9mettagrid_11mettagrid_c_MettaGrid, __pyx_n_s_step, __pyx_t_4) < 0) __PYX_ERR(0, 200, __pyx_L1_error)
+  if (__Pyx_SetItemOnTypeDict((PyObject *)__pyx_ptype_9mettagrid_11mettagrid_c_MettaGrid, __pyx_n_s_step, __pyx_t_4) < 0) __PYX_ERR(0, 209, __pyx_L1_error)
   __Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
   PyType_Modified(__pyx_ptype_9mettagrid_11mettagrid_c_MettaGrid);
 
-  /* "mettagrid/mettagrid.pyx":226
+  /* "mettagrid/mettagrid.pyx":235
  *         return (obs, rewards, terms, truncs, infos)
  * 
  *     cpdef dict get_episode_stats(self):             # <<<<<<<<<<<<<<
  *         return {
  *             "game": self._stats.stats(),
  */
-  __pyx_t_4 = __Pyx_CyFunction_New(&__pyx_mdef_9mettagrid_11mettagrid_c_9MettaGrid_11get_episode_stats, __Pyx_CYFUNCTION_CCLASS, __pyx_n_s_MettaGrid_get_episode_stats, NULL, __pyx_n_s_mettagrid_mettagrid_c, __pyx_d, ((PyObject *)__pyx_codeobj__32)); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 226, __pyx_L1_error)
+  __pyx_t_4 = __Pyx_CyFunction_New(&__pyx_mdef_9mettagrid_11mettagrid_c_9MettaGrid_11get_episode_stats, __Pyx_CYFUNCTION_CCLASS, __pyx_n_s_MettaGrid_get_episode_stats, NULL, __pyx_n_s_mettagrid_mettagrid_c, __pyx_d, ((PyObject *)__pyx_codeobj__32)); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 235, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_4);
-  if (__Pyx_SetItemOnTypeDict((PyObject *)__pyx_ptype_9mettagrid_11mettagrid_c_MettaGrid, __pyx_n_s_get_episode_stats, __pyx_t_4) < 0) __PYX_ERR(0, 226, __pyx_L1_error)
+  if (__Pyx_SetItemOnTypeDict((PyObject *)__pyx_ptype_9mettagrid_11mettagrid_c_MettaGrid, __pyx_n_s_get_episode_stats, __pyx_t_4) < 0) __PYX_ERR(0, 235, __pyx_L1_error)
   __Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
   PyType_Modified(__pyx_ptype_9mettagrid_11mettagrid_c_MettaGrid);