Skip to content

Commit e32edd8

Browse files
committed
fix: wrap data_home in Path()
Closes #39
1 parent 3bc275b commit e32edd8

1 file changed

Lines changed: 4 additions & 6 deletions

File tree

libsvmdata/abstraction.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,9 @@ def _get_data_home(subdir_name=""):
1818
data_home = os.environ.get("XDG_DATA_HOME", None)
1919
if data_home is None:
2020
data_home = Path.home() / "data"
21-
return data_home / subdir_name
21+
else:
22+
data_home = Path(data_home)
23+
return data_home / subdir_name if subdir_name else data_home
2224

2325

2426
class AbstractDataset(ABC):
@@ -86,7 +88,6 @@ def get_X_y(self, replace=False, verbose=False):
8688
exists = False
8789

8890
if replace or not exists:
89-
9091
# Remove existing dataset files if there are any
9192
if raw_dataset_path.exists():
9293
raw_dataset_path.unlink()
@@ -106,10 +107,7 @@ def get_X_y(self, replace=False, verbose=False):
106107

107108
if verbose:
108109
print("Loading file and saving data...")
109-
X, y = self._load_file_and_save_data(
110-
raw_dataset_path,
111-
ext_dataset_path
112-
)
110+
X, y = self._load_file_and_save_data(raw_dataset_path, ext_dataset_path)
113111

114112
else:
115113
if verbose:

0 commit comments

Comments
 (0)