frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

torch.hub: add get_dir/set_dir type hints (#134906)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/134906
Approved by: https://github.com/Skylion007
This commit is contained in:
Adam J. Stewart
2024-09-12 03:53:27 +00:00
committed by PyTorch MergeBot
parent 79223114db
commit 5bc238c73e
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
torch/hub.py
View File

@ -12,7 +12,7 @@ import uuid
import warnings
import zipfile
from pathlib import Path
from typing import Any, Dict, Optional
from typing import Any, Dict, Optional, Union
from typing_extensions import deprecated
from urllib.error import HTTPError, URLError
from urllib.parse import urlparse # noqa: F401
@ -389,7 +389,7 @@ def _load_entry_from_hubconf(m, model):
return func
def get_dir():
def get_dir() -> str:
r"""
Get the Torch Hub cache directory used for storing downloaded models & weights.
@ -408,7 +408,7 @@ def get_dir():
return os.path.join(_get_torch_home(), "hub")
def set_dir(d):
def set_dir(d: Union[str, os.PathLike]) -> None:
r"""
Optionally set the Torch Hub directory used to save downloaded models & weights.