Force data types for TF compatibility

Feryal · Jul 10, 2018 · c38c0bb · c38c0bb
1 parent aaaf377
commit c38c0bb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/craft.py b/craft.py
@@ -234,7 +234,7 @@ def features(self):
             assert len(features) == self.world.n_features
             self._cached_features = features
 
-        return self._cached_features
+        return self._cached_features.astype(np.float32)
 
     def step(self, action):
         x, y = self.pos

diff --git a/env.py b/env.py
@@ -54,7 +54,7 @@ def step(self, action, num_steps=1):
         state_reward, self._current_state = self._current_state.step(action)
 
         done = self._isDone()
-        reward = self._getReward() + state_reward
+        reward = np.float32(self._getReward() + state_reward)
 
         if done:
             self.reset()