This is an automated email from the ASF dual-hosted git repository. hui pushed a commit to branch mlnode/test in repository https://gitbox.apache.org/repos/asf/iotdb.git
commit 497af731998dd23d7e2dc0bcd0cf3dedd2225db4 Author: Minghui Liu <[email protected]> AuthorDate: Tue Apr 4 17:08:11 2023 +0800 fix some bug --- mlnode/iotdb/mlnode/exception.py | 6 +++++- mlnode/iotdb/mlnode/parser.py | 2 +- mlnode/iotdb/mlnode/service.py | 5 ++--- 3 files changed, 8 insertions(+), 5 deletions(-) diff --git a/mlnode/iotdb/mlnode/exception.py b/mlnode/iotdb/mlnode/exception.py index be601794c9..47edb95eb4 100644 --- a/mlnode/iotdb/mlnode/exception.py +++ b/mlnode/iotdb/mlnode/exception.py @@ -18,7 +18,11 @@ class _BaseError(Exception): """Base class for exceptions in this module.""" - pass + def __init__(self): + self.message = None + + def __str__(self) -> str: + return self.message class BadNodeUrlError(_BaseError): diff --git a/mlnode/iotdb/mlnode/parser.py b/mlnode/iotdb/mlnode/parser.py index ebcd97dee6..7d920aa84f 100644 --- a/mlnode/iotdb/mlnode/parser.py +++ b/mlnode/iotdb/mlnode/parser.py @@ -95,7 +95,7 @@ _data_config_parser.add_argument('--source_type', type=DataSourceType, default=DataSourceType.THRIFT, choices=list(DataSourceType)) -_data_config_parser.add_argument('--dataset_type', required=True, +_data_config_parser.add_argument('--dataset_type', type=DatasetType, default=DatasetType.WINDOW, choices=list(DatasetType)) diff --git a/mlnode/iotdb/mlnode/service.py b/mlnode/iotdb/mlnode/service.py index ae0727cc5a..8bf97402d1 100644 --- a/mlnode/iotdb/mlnode/service.py +++ b/mlnode/iotdb/mlnode/service.py @@ -19,7 +19,7 @@ import threading import time from thrift.protocol import TCompactProtocol -from thrift.server import TProcessPoolServer +from thrift.server import TServer from thrift.transport import TSocket, TTransport from iotdb.mlnode.config import descriptor @@ -37,8 +37,7 @@ class RPCService(threading.Thread): transport_factory = TTransport.TFramedTransportFactory() protocol_factory = TCompactProtocol.TCompactProtocolFactory() - self.__pool_server = TProcessPoolServer.TProcessPoolServer(processor, transport, transport_factory, - protocol_factory) + self.__pool_server = TServer.TThreadPoolServer(processor, transport, transport_factory, protocol_factory) def run(self) -> None: logger.info("The RPC service thread begin to run...")
