Skip to content

Commit cb91cbf

Browse files
authored
[AINode] Update python dependencies (#15712)
1 parent 4e2fb58 commit cb91cbf

File tree

3 files changed

+99
-46
lines changed

3 files changed

+99
-46
lines changed

iotdb-core/ainode/ainode/core/handler.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,8 @@
1616
# under the License.
1717
#
1818

19+
from ainode.core.constant import TSStatusCode
20+
from ainode.core.log import Logger
1921
from ainode.core.manager.cluster_manager import ClusterManager
2022
from ainode.core.manager.inference_manager import InferenceManager
2123
from ainode.core.manager.model_manager import ModelManager
@@ -33,6 +35,8 @@
3335
)
3436
from ainode.thrift.common.ttypes import TSStatus
3537

38+
logger = Logger()
39+
3640

3741
class AINodeRPCServiceHandler(IAINodeRPCService.Iface):
3842
def __init__(self):

iotdb-core/ainode/poetry.lock

Lines changed: 93 additions & 45 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

iotdb-core/ainode/pyproject.toml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ packages = [
4949
python = ">=3.9, <3.13"
5050
numpy = "^1.21.4"
5151
pandas = "^1.3.5"
52-
torch = ">=2.2.0"
52+
torch = "=2.4.0"
5353
pylru = "^1.2.1"
5454
thrift = ">=0.14.0"
5555
dynaconf = "^3.1.11"
@@ -66,6 +66,7 @@ huggingface_hub = "^0.30.1"
6666
black = "25.1.0"
6767
isort = "6.0.1"
6868
transformers = "==4.40.1"
69+
torchmetrics = ">=1.2.0"
6970

7071
[tool.poetry.scripts]
7172
ainode = "ainode.core.script:main"

0 commit comments

Comments
 (0)