Skip to content

Commit cccc888

Browse files
committed
fixes #180
1 parent 60e55a6 commit cccc888

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

visual_behavior/translator/foraging/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -187,7 +187,7 @@ def stringify(x):
187187
for col in forced_string:
188188
trials[col] = trials[col].map(stringify)
189189

190-
trials['change_frame'] = trials['change_frame'].map(lambda x: int(x) if np.isfinite(x) else None)
190+
trials['change_frame'] = trials['change_frame'].map(lambda x: int(x) if np.isfinite(x) else None) + 1
191191
trials["change_image_category"] = trials["change_image_category"].apply(lambda x: x if x else '') # use empty string instead of NoneType
192192
trials["change_image_name"] = trials["change_image_name"].apply(lambda x: x if x else '') # use empty string instead of NoneType
193193
trials["initial_image_category"] = trials["initial_image_category"].apply(lambda x: x if x else '') # use empty string instead of NoneType

0 commit comments

Comments
 (0)