[frontend util] fix trial duration estimation

This commit is contained in:
Jan Grewe 2020-08-14 13:39:07 +02:00
parent 94221463f2
commit 0da2bfa110

View File

@ -207,7 +207,7 @@ class StimSpikesFile:
for l in lines: for l in lines:
l = l.strip() l = l.strip()
if "duration:" in l: if "duration:" in l:
trial_duration = float(l[1:].strip().split(":")[-3]) trial_duration = float(l[1:].strip().split(":")[-1][:-3])
if "index:" in l: if "index:" in l:
if len(trial_data) > 0: if len(trial_data) > 0:
index_map[(index, trial)] = (trial_duration, trial_data) index_map[(index, trial)] = (trial_duration, trial_data)
@ -226,5 +226,5 @@ class StimSpikesFile:
def get(self, run_index, trial_index): def get(self, run_index, trial_index):
if tuple([run_index, trial_index]) not in self._data_map.keys(): if tuple([run_index, trial_index]) not in self._data_map.keys():
print("Data not found for run %i and trial %i:" % (run_index, trial_index)) print("Data not found for run %i and trial %i:" % (run_index, trial_index))
return None return None, None
return self._data_map[(run_index, trial_index)] return self._data_map[(run_index, trial_index)]