Commits

dan mackinlay committed 0d70b1b

re-commit fix to graph stats

Comments (0)

Files changed (1)

             self.max_state_array_floats,
             num_agents, dimensions, keep_steps
             ))
-        raw_output_vels = np.zeros((keep_steps, num_agents, dimensions))
-        raw_output_locs = np.zeros((keep_steps, num_agents, dimensions))
+        raw_output_vels = np.zeros((keep_steps, num_agents, dimensions),
+            dtype=np.float32)
+        raw_output_locs = np.zeros((keep_steps, num_agents, dimensions),
+            dtype=np.float32)
         cum_output_adjs = np.zeros((num_agents, num_agents),
-           dtype=np.int)
+            dtype=np.int32)
         
         self.raw_output_vels = raw_output_vels
         self.raw_output_locs = raw_output_locs
         
         for i in xrange(keep_steps):
             self.tick()
+            #why are these steps not in the main tick loop?
             raw_output_vels[i] = self.vels
-            raw_output_locs[i] = self.locs
-            cum_output_adjs += self.neighbours
+            raw_output_locs[i] = self.locs 
+            self.cum_output_adjs += self.neighbours
     
     def __getstate__(self):
         state = self.__dict__.copy()