Added remove stimuli with no fixation filter to dataset_config.py (#44)

* Added filter to remove stimuli with no fixations * Update test_filter_datasets.py * Update test_filter_datasets.py * Added remove stimuli without fixations to dataset_config.py
matthias-k · Dec 16, 2023 · e32e2af · e32e2af
1 parent 90d6a99
commit e32e2af
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pysaliency/dataset_config.py b/pysaliency/dataset_config.py
@@ -9,7 +9,8 @@
     filter_scanpaths_by_attribute,
     filter_fixations_by_attribute,
     filter_stimuli_by_attribute,
-    filter_scanpaths_by_length
+    filter_scanpaths_by_length,
+    remove_stimuli_without_fixations
 )
 
 from schema import Schema, Optional
@@ -49,7 +50,8 @@ def apply_dataset_filter_config(stimuli, fixations, filter_config):
         'filter_scanpaths_by_attribute': add_stimuli_argument(filter_scanpaths_by_attribute),
         'filter_fixations_by_attribute': add_stimuli_argument(filter_fixations_by_attribute),
         'filter_stimuli_by_attribute': filter_stimuli_by_attribute,
-        'filter_scanpaths_by_length': add_stimuli_argument(filter_scanpaths_by_length)
+        'filter_scanpaths_by_length': add_stimuli_argument(filter_scanpaths_by_length),
+        'remove_stimuli_without_fixations': remove_stimuli_without_fixations
     }
 
     if filter_config['type'] not in filter_dict: