recorder fixed

environments/factory/additional/item/factory_item.py

@@ -3,7 +3,7 @@ from typing import List, Union, Dict
 import numpy as np
 import random
 
-from environments.factory.additional.item.item_collections import ItemRegister, Inventories, DropOffLocations
+from environments.factory.additional.item.item_collections import Items, Inventories, DropOffLocations
 from environments.factory.additional.item.item_util import Constants, Actions, RewardsItem, ItemProperties
 from environments.factory.base.base_factory import BaseFactory
 from environments.factory.base.objects import Agent, Action
@@ -49,7 +49,7 @@ class ItemFactory(BaseFactory):
             entity_kwargs=dict(
                 storage_size_until_full=self.item_prop.max_dropoff_storage_size)
         )
-        item_register = ItemRegister(self._level_shape)
+        item_register = Items(self._level_shape)
         empty_tiles = self[c.FLOOR].empty_tiles[:self.item_prop.n_items]
         item_register.spawn_items(empty_tiles)
 

environments/factory/additional/item/item_collections.py

@@ -7,7 +7,7 @@ from environments.factory.base.registers import EntityCollection, BoundEnvObjCol
 from environments.factory.additional.item.item_entities import Item, DropOffLocation
 
 
-class ItemRegister(EntityCollection):
+class Items(EntityCollection):
 
     _accepted_objects = Item
 
@@ -37,9 +37,9 @@ class Inventory(BoundEnvObjCollection):
         return super(Inventory, self).as_array()
 
     def summarize_states(self, **kwargs):
-        attr_dict = {key: str(val) for key, val in self.__dict__.items() if not key.startswith('_') and key != 'data'}
-        attr_dict.update(dict(items={key: val.summarize_state(**kwargs) for key, val in self.items()}))
-        attr_dict.update(dict(name=self.name))
+        attr_dict = {key: val for key, val in self.__dict__.items() if not key.startswith('_') and key != 'data'}
+        attr_dict.update(dict(items=[val.summarize_state(**kwargs) for key, val in self.items()]))
+        attr_dict.update(dict(name=self.name, belongs_to=self._bound_entity.name))
         return attr_dict
 
     def pop(self):
@@ -79,9 +79,11 @@ class Inventories(ObjectCollection):
             return None
 
     def summarize_states(self, **kwargs):
-        return {key: val.summarize_states(**kwargs) for key, val in self.items()}
+        return [val.summarize_states(**kwargs) for key, val in self.items()]
 
 
 class DropOffLocations(EntityCollection):
 
     _accepted_objects = DropOffLocation
+    _stateless_entities = True
+

environments/factory/additional/item/item_entities.py

@@ -26,7 +26,7 @@ class Item(Entity):
     def set_tile_to(self, no_pos_tile):
         self._tile = no_pos_tile
 
-    def summarize_state(self, **_) -> dict:
+    def summarize_state(self) -> dict:
         super_summarization = super(Item, self).summarize_state()
         super_summarization.update(dict(auto_despawn=self.auto_despawn))
         return super_summarization
@@ -55,7 +55,3 @@ class DropOffLocation(Entity):
     @property
     def is_full(self):
         return False if not self.storage.maxlen else self.storage.maxlen == len(self.storage)
-
-    def summarize_state(self, n_steps=None) -> dict:
-        if n_steps == h.STEPS_START:
-            return super().summarize_state(n_steps=n_steps)