mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-11-02 23:41:27 +00:00
36 lines
888 B
Python
36 lines
888 B
Python
import importlib
|
|
|
|
import importlib_metadata
|
|
from PyInstaller.utils.hooks import copy_metadata
|
|
|
|
datas = copy_metadata("transformers")
|
|
datas += copy_metadata("tokenizers")
|
|
datas += copy_metadata("tqdm")
|
|
datas += copy_metadata("regex")
|
|
datas += copy_metadata("requests")
|
|
datas += copy_metadata("packaging")
|
|
datas += copy_metadata("filelock")
|
|
datas += copy_metadata("numpy")
|
|
datas += copy_metadata("torch")
|
|
|
|
candidates = [
|
|
"tensorflow",
|
|
"tensorflow-cpu",
|
|
"tensorflow-gpu",
|
|
"tf-nightly",
|
|
"tf-nightly-cpu",
|
|
"tf-nightly-gpu",
|
|
"intel-tensorflow",
|
|
"intel-tensorflow-avx512",
|
|
"tensorflow-rocm",
|
|
"tensorflow-macos",
|
|
"tensorflow-aarch64",
|
|
]
|
|
for candidate in candidates:
|
|
try:
|
|
if importlib.util.find_spec(candidate):
|
|
datas += copy_metadata(candidate)
|
|
break
|
|
except importlib_metadata.PackageNotFoundError:
|
|
pass
|