Update salicon.py

matthias-k · Nov 28, 2024 · 241855b · 241855b
1 parent e962c0a
commit 241855b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pysaliency/external_datasets/salicon.py b/pysaliency/external_datasets/salicon.py
@@ -253,8 +253,8 @@ def get_SALICON_test(edition='2015', fixation_type='mouse', location=None):
     if location:
         name = _get_SALICON_name(edition=edition, fixation_type=fixation_type)
         if os.path.exists(os.path.join(location, name)):
-            stimuli = _load(os.path.join('SALICON', 'stimuli_test.hdf5'))
+            stimuli = _load(os.path.join(location, 'SALICON', 'stimuli_test.hdf5'))
             return stimuli
     _, _, stimuli_test, _, _ = get_SALICON(location=location, edition=edition, fixation_type=fixation_type)
 
-    return stimuli_test
+    return stimuli_test