Skip to content

Commit 8e8cd6f

Browse files
committed
spotless ainode codes
1 parent 8d00ce7 commit 8e8cd6f

File tree

4 files changed

+12
-4
lines changed

4 files changed

+12
-4
lines changed

iotdb-core/ainode/iotdb/ainode/core/inference/inference_request_pool.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,11 @@
3030
from iotdb.ainode.core.constant import INFERENCE_LOG_FILE_NAME_PREFIX_TEMPLATE
3131
from iotdb.ainode.core.inference.batcher.basic_batcher import BasicBatcher
3232
from iotdb.ainode.core.inference.inference_request import InferenceRequest
33-
from iotdb.ainode.core.inference.pipeline.basic_pipeline import ForecastPipeline, ClassificationPipeline, ChatPipeline
33+
from iotdb.ainode.core.inference.pipeline.basic_pipeline import (
34+
ChatPipeline,
35+
ClassificationPipeline,
36+
ForecastPipeline,
37+
)
3438
from iotdb.ainode.core.inference.pipeline.pipeline_loader import load_pipeline
3539
from iotdb.ainode.core.inference.request_scheduler.basic_request_scheduler import (
3640
BasicRequestScheduler,

iotdb-core/ainode/iotdb/ainode/core/manager/inference_manager.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,8 +34,12 @@
3434
InferenceRequest,
3535
InferenceRequestProxy,
3636
)
37+
from iotdb.ainode.core.inference.pipeline.basic_pipeline import (
38+
ChatPipeline,
39+
ClassificationPipeline,
40+
ForecastPipeline,
41+
)
3742
from iotdb.ainode.core.inference.pipeline.pipeline_loader import load_pipeline
38-
from iotdb.ainode.core.inference.pipeline.basic_pipeline import ForecastPipeline, ClassificationPipeline, ChatPipeline
3943
from iotdb.ainode.core.inference.pool_controller import PoolController
4044
from iotdb.ainode.core.inference.utils import generate_req_id
4145
from iotdb.ainode.core.log import Logger

iotdb-core/ainode/iotdb/ainode/core/model/sundial/pipeline_sundial.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,8 @@
1818

1919
import torch
2020

21-
from iotdb.ainode.core.inference.pipeline.basic_pipeline import ForecastPipeline
2221
from iotdb.ainode.core.exception import InferenceModelInternalError
22+
from iotdb.ainode.core.inference.pipeline.basic_pipeline import ForecastPipeline
2323

2424

2525
class SundialPipeline(ForecastPipeline):

iotdb-core/ainode/iotdb/ainode/core/model/timer_xl/pipeline_timer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,8 @@
1818

1919
import torch
2020

21-
from iotdb.ainode.core.inference.pipeline.basic_pipeline import ForecastPipeline
2221
from iotdb.ainode.core.exception import InferenceModelInternalError
22+
from iotdb.ainode.core.inference.pipeline.basic_pipeline import ForecastPipeline
2323

2424

2525
class TimerPipeline(ForecastPipeline):

0 commit comments

Comments
 (0)