Merge pull request #30 from BIDMCDigitalPsychiatry/avaidyam-patch-1

Add PyYAML to the dependencies
BIDMCDigitalPsychiatry · May 21, 2021 · 3744de8 · 3744de8
2 parents f27fc03 + da8f03d
commit 3744de8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/cortex/feature_types.py b/cortex/feature_types.py
@@ -60,7 +60,7 @@ def _wrapper2(*args, **kwargs):
                 elif os.getenv('CORTEX_CACHE_DIR') is not None:
                     cache_dir = os.path.expanduser(os.getenv('CORTEX_CACHE_DIR'))
                     assert os.path.exists(cache_dir), f"Caching directory ({cache_dir}) found in enviornmental variables does not exist"
-                if cache_dir is None: 
+                else:
                     cache_dir = os.path.expanduser('~/.cache/cortex')
                     if not os.path.exists(cache_dir):
                         log.info(f"Caching directory does not yet exist, creating...")

diff --git a/pyproject.toml b/pyproject.toml
@@ -33,6 +33,7 @@ fastdtw = "^0.3.4"
 sklearn = "^0.0"
 pytz = "^2021.1"
 compress-pickle = "^2.0.1"
+pyyaml = "^5.4.1"
 
 [tool.poetry.dev-dependencies]