Skip to content

Commit 3deb1e5

Browse files
committed
Debug and update unit tests.
1 parent def3732 commit 3deb1e5

File tree

2 files changed

+7
-2
lines changed

2 files changed

+7
-2
lines changed

allensdk/brain_observatory/behavior/data_objects/stimuli/presentations.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -261,8 +261,9 @@ def from_nwb(
261261
stimulus_presentations=table
262262
)
263263
trials = None
264-
if add_trials_dependent_values:
264+
if add_trials_dependent_values and nwbfile.trials is not None:
265265
trials = Trials.from_nwb(nwbfile)
266+
266267
return Presentations(presentations=table,
267268
column_list=column_list,
268269
trials=trials)
@@ -420,7 +421,9 @@ def from_stimulus_file(
420421
)
421422

422423
return Presentations(
423-
presentations=stim_pres_df, column_list=column_list
424+
presentations=stim_pres_df,
425+
column_list=column_list,
426+
trials=trials
424427
)
425428

426429
@classmethod

allensdk/test/brain_observatory/behavior/data_objects/test_stimuli.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,8 @@ def data(self):
5252
{
5353
"start_time": [300.0, 330.0, 360.0],
5454
"stop_time": [330.0, 360.0, 360.0],
55+
"catch": [False, True, False],
56+
"change_frame": [-99, 99, -99]
5557
}
5658
)
5759

0 commit comments

Comments
 (0)