| 1234567891011121314151617181920 |
- import os
- import site
- import sys
- import typing
- import torch
- def _prefix_regex() -> typing.List[str]:
- raw_paths = (
- site.getsitepackages()
- + sys.path
- + [site.getuserbase()]
- + [site.getusersitepackages()]
- + [os.path.dirname(os.path.dirname(torch.__file__))]
- )
- path_prefixes = sorted({os.path.abspath(i) for i in raw_paths}, reverse=True)
- assert all(isinstance(i, str) for i in path_prefixes)
- return [i + os.sep for i in path_prefixes]
|
