Skip to content

Commit ea4d6c4

Browse files
olichejuhuntenburg
authored andcommitted
Session loader: cast wheel data as float32
1 parent ff64d41 commit ea4d6c4

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

brainbox/io/one.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1240,6 +1240,7 @@ def load_wheel(self, sampling_rate=1000, smooth_size=0.03):
12401240
wheel_raw['timestamps'], wheel_raw['position'], freq=sampling_rate)
12411241
self.wheel['velocity'], self.wheel['acceleration'] = velocity_smoothed(
12421242
self.wheel['position'], freq=sampling_rate, smooth_size=smooth_size)
1243+
self.wheel = self.wheel.apply(np.float32)
12431244
self.data_info.loc[self.data_info['name'] == 'wheel', 'is_loaded'] = True
12441245

12451246
def load_pose(self, likelihood_thr=0.9, views=['left', 'right', 'body']):

0 commit comments

Comments
 (0)