simplify RESULTS_DIR

microsoft · Jan 7, 2025 · 87db7f8 · 87db7f8
1 parent b8e0b7c
commit 87db7f8
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/notebooks_stories/2_analyze/flatmaps_helper.py b/notebooks_stories/2_analyze/flatmaps_helper.py
@@ -2,13 +2,12 @@
 import joblib
 import numpy as np
 import sys
-import os
 from os.path import abspath, dirname, join
-# try:
-# from sasc.config import FMRI_DIR, STORIES_DIR, RESULTS_DIR, CACHE_DIR, cache_ngrams_dir, regions_idxs_dir
-# except ImportError:
-repo_path = dirname(dirname(dirname(abspath(__file__))))
-RESULTS_DIR = join(repo_path, 'results')
+try:
+    from sasc.config import FMRI_DIR, STORIES_DIR, RESULTS_DIR, CACHE_DIR, cache_ngrams_dir, regions_idxs_dir
+except ImportError:
+    repo_path = dirname(dirname(dirname(abspath(__file__))))
+    RESULTS_DIR = join(repo_path, 'results')
 
 sys.path.append('../notebooks')
 VOX_COUNTS = {