Skip to content

Commit 305409d

Browse files
committed
update dependency collection in .spec
1 parent 08a0ea6 commit 305409d

File tree

2 files changed

+8
-5
lines changed

2 files changed

+8
-5
lines changed

iotdb-core/ainode/ainode.spec

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -41,9 +41,12 @@ all_hiddenimports = []
4141
# Only collect essential data files and binaries for critical libraries
4242
# This reduces startup time by avoiding unnecessary module imports
4343
essential_libraries = {
44-
'torch': True, # Keep collect_all for torch as it has many dynamic imports
45-
'transformers': True, # Keep collect_all for transformers
46-
'safetensors': True, # Keep collect_all for safetensors
44+
'torch': True,
45+
'transformers': True,
46+
'tokenizers': True,
47+
'huggingface_hub': True,
48+
'safetensors': True,
49+
'hf_xet': True,
4750
'numpy': True,
4851
'scipy': True,
4952
'pandas': True,

iotdb-core/ainode/pyproject.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ exclude = [
7979
python = ">=3.11.0,<3.14.0"
8080

8181
# ---- DL / HF stack ----
82-
torch = "^2.7.1,<2.8.0"
82+
torch = "^2.7.2,<2.8.0"
8383
torchmetrics = "^1.8.0"
8484
transformers = "==4.56.2"
8585
tokenizers = ">=0.22.0,<=0.23.0"
@@ -115,7 +115,7 @@ black = "25.1.0"
115115
isort = "6.0.1"
116116
setuptools = ">=75.3.0"
117117
joblib = ">=1.4.2"
118-
urllib3 = ">=2.2.3"
118+
urllib3 = "^2.5.0"
119119

120120
[tool.poetry.scripts]
121121
ainode = "iotdb.ainode.core.script:main"

0 commit comments

Comments
 (0)