diff --git a/fishbook/frontend/util.py b/fishbook/frontend/util.py index b8e2665..234dee0 100644 --- a/fishbook/frontend/util.py +++ b/fishbook/frontend/util.py @@ -207,7 +207,16 @@ class StimSpikesFile: for l in lines: l = l.strip() if "duration:" in l: - trial_duration = float(l[1:].strip().split(":")[-1][:-3]) + trial_duration = l[1:].strip().split(":")[-1].strip() + if "sec" in trial_duration: + trial_duration = float(trial_duration[:-3]) + elif "ms" in trial_duration: + trial_duration = float(trial_duration[:-2]) / 1000 + elif trial_duration[-1] == 's': + trial_duration = float(trial_duration[:1]) + else: + trial_duration = 0.0 + print(l, trial_duration) if "index:" in l: if len(trial_data) > 0: index_map[(index, trial)] = (trial_duration, trial_data)