Lab state fix - running now.

michaelschaarschmidt · michaelschaarschmidt · commit faaf0851a09d · 2017-07-02T10:09:40.000+01:00
diff --git a/examples/lab_main.py b/examples/lab_main.py
@@ -90,11 +90,8 @@ def main():
 
     if args.debug:
         logger.info("-" * 16)
-        logger.info("Agent configuration:")
-        logger.info(agent.config)
-        if agent.model:
-            logger.info("Model configuration:")
-            logger.info(agent.model.config)
+        logger.info("Configuration:")
+        logger.info(agent_config)
 
     runner = Runner(
         agent=agent,
diff --git a/tensorforce/environments/deepmind_lab.py b/tensorforce/environments/deepmind_lab.py
@@ -111,10 +111,10 @@ def states(self):
             state_type = state['dtype']
 
             if state_type == np.uint8:
-                state_type = np.int32
+                state_type = np.float32
 
             if state['name'] == self.state_attribute:
-                states[state['name']] = dict(shape=state['shape'], type=state_type)
+                return dict(shape=state['shape'], type=state_type)
 
         return states
 
