From 4e7e93e259501d315063c871e0a1912dd5c45f5a Mon Sep 17 00:00:00 2001 From: Parmeet Singh Bhatia Date: Wed, 25 May 2022 12:40:52 -0400 Subject: [PATCH] Change root directory for datasets --- torchtext/__init__.py | 2 +- torchtext/data/datasets_utils.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/torchtext/__init__.py b/torchtext/__init__.py index 70dcb0d0c3..81ba54acd2 100644 --- a/torchtext/__init__.py +++ b/torchtext/__init__.py @@ -4,7 +4,7 @@ from torchtext import _extension # noqa: F401 _TEXT_BUCKET = "https://download.pytorch.org/models/text/" -_CACHE_DIR = os.path.expanduser("~/.torchtext/cache") +_CACHE_DIR = os.path.expanduser("~/.cache/torch/text") from . import data, datasets, experimental, functional, models, nn, transforms, utils, vocab diff --git a/torchtext/data/datasets_utils.py b/torchtext/data/datasets_utils.py index d8628517cb..0c39751176 100644 --- a/torchtext/data/datasets_utils.py +++ b/torchtext/data/datasets_utils.py @@ -187,7 +187,7 @@ def decorator(fn): @functools.wraps(fn) def wrapper(root=_CACHE_DIR, *args, **kwargs): - new_root = os.path.join(root, dataset_name) + new_root = os.path.join(root, "datasets", dataset_name) if not os.path.exists(new_root): os.makedirs(new_root, exist_ok=True) return fn(root=new_root, *args, **kwargs)