1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
import os
import os.path as osp
from typing import Optional
ENV_PYG_HOME = 'PYG_HOME'
DEFAULT_CACHE_DIR = osp.join('~', '.cache', 'pyg')
_home_dir: Optional[str] = None
def get_home_dir() -> str:
r"""Get the cache directory used for storing all :pyg:`PyG`-related data.
If :meth:`set_home_dir` is not called, the path is given by the environment
variable :obj:`$PYG_HOME` which defaults to :obj:`"~/.cache/pyg"`.
"""
if _home_dir is not None:
return _home_dir
return osp.expanduser(os.getenv(ENV_PYG_HOME, DEFAULT_CACHE_DIR))
def set_home_dir(path: str) -> None:
r"""Set the cache directory used for storing all :pyg:`PyG`-related data.
Args:
path (str): The path to a local folder.
"""
global _home_dir
_home_dir = path
|
