mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
5fd881a5b6
This reverts commit 54a00af2c6026a830f40d9e6a659ff81d51f9bc6. Reverted https://github.com/pytorch/pytorch/pull/145175 on behalf of https://github.com/huydhn due to Sorry for reverting your change but it seems to break some trunk tests ([comment](https://github.com/pytorch/pytorch/pull/145175#issuecomment-2603418267))
21 lines
497 B
Python
21 lines
497 B
Python
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]
|