new plotting, omit_agent_obs

2021-06-02 18:12:56 +02:00
parent 8810955e86
commit b72013407e
5 changed files with 79 additions and 20 deletions
--- a/environments/factory/simple_factory.py
+++ b/environments/factory/simple_factory.py
@ -37,8 +37,7 @@ class SimpleFactory(BaseFactory):
        self.dirt_properties = dirt_properties
        self.verbose = verbose
        self.max_dirt = 20
-        super(SimpleFactory, self).__init__(*args, **kwargs)
-        self._state_slices.register_additional_items('dirt')
+        super(SimpleFactory, self).__init__(*args, additional_slices='dirt', **kwargs)
        self._renderer = None  # expensive - don't use it when not required !

    def render(self):