From a9efdabe75eb30ace7f131c412d92b6a5e7a2994 Mon Sep 17 00:00:00 2001 From: Harneet Singh Khanuja Date: Mon, 6 Nov 2023 19:16:17 +0100 Subject: [PATCH] Update dataset_config.py --- pysaliency/dataset_config.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pysaliency/dataset_config.py b/pysaliency/dataset_config.py index a94148c..504f532 100644 --- a/pysaliency/dataset_config.py +++ b/pysaliency/dataset_config.py @@ -5,7 +5,7 @@ filter_stimuli_by_size, train_split, validation_split, - test_split, + test_split ) from schema import Schema, Optional @@ -51,7 +51,7 @@ def apply_dataset_filter_config(stimuli, fixations, filter_config): return filter_fn(stimuli, fixations, **filter_config['parameters']) - + def _clip_out_of_stimulus_fixations(stimuli, fixations): clipped_fixations = clip_out_of_stimulus_fixations(fixations, stimuli=stimuli) return stimuli, clipped_fixations