diff --git a/libsvmdata/abstraction.py b/libsvmdata/abstraction.py index cefc30d..467df5f 100644 --- a/libsvmdata/abstraction.py +++ b/libsvmdata/abstraction.py @@ -14,10 +14,14 @@ def _get_data_home(subdir_name=""): seek for the variable $XDG_DATA_HOME. Finally, the fallback is $HOME/data. """ data_home = os.environ.get("LIBSVMDATA_HOME", None) + if data_home is None: data_home = os.environ.get("XDG_DATA_HOME", None) if data_home is None: data_home = Path.home() / "data" + else: + data_home = Path(data_home) + return data_home / subdir_name