From 3701078242542819e54182a0ba7b03d44b5a1163 Mon Sep 17 00:00:00 2001 From: Liu Zhengyun Date: Fri, 9 Jan 2026 17:32:27 +0800 Subject: [PATCH] fix pipeline bug --- .../iotdb/ainode/core/inference/pipeline/basic_pipeline.py | 6 +++--- .../iotdb/ainode/core/model/chronos2/pipeline_chronos2.py | 2 +- .../iotdb/ainode/core/model/sktime/pipeline_sktime.py | 2 +- .../iotdb/ainode/core/model/sundial/pipeline_sundial.py | 2 +- .../iotdb/ainode/core/model/timer_xl/pipeline_timer.py | 2 +- 5 files changed, 7 insertions(+), 7 deletions(-) diff --git a/iotdb-core/ainode/iotdb/ainode/core/inference/pipeline/basic_pipeline.py b/iotdb-core/ainode/iotdb/ainode/core/inference/pipeline/basic_pipeline.py index f1704fb90c4c..f37d617748aa 100644 --- a/iotdb-core/ainode/iotdb/ainode/core/inference/pipeline/basic_pipeline.py +++ b/iotdb-core/ainode/iotdb/ainode/core/inference/pipeline/basic_pipeline.py @@ -48,7 +48,7 @@ def postprocess(self, outputs, **infer_kwargs): class ForecastPipeline(BasicPipeline): def __init__(self, model_info: ModelInfo, **model_kwargs): - super().__init__(model_info, model_kwargs=model_kwargs) + super().__init__(model_info, **model_kwargs) def preprocess( self, @@ -199,7 +199,7 @@ def postprocess( class ClassificationPipeline(BasicPipeline): def __init__(self, model_info: ModelInfo, **model_kwargs): - super().__init__(model_info, model_kwargs=model_kwargs) + super().__init__(model_info, **model_kwargs) def preprocess(self, inputs, **kwargs): return inputs @@ -214,7 +214,7 @@ def postprocess(self, outputs, **kwargs): class ChatPipeline(BasicPipeline): def __init__(self, model_info: ModelInfo, **model_kwargs): - super().__init__(model_info, model_kwargs=model_kwargs) + super().__init__(model_info, **model_kwargs) def preprocess(self, inputs, **kwargs): return inputs diff --git a/iotdb-core/ainode/iotdb/ainode/core/model/chronos2/pipeline_chronos2.py b/iotdb-core/ainode/iotdb/ainode/core/model/chronos2/pipeline_chronos2.py index 3fdc7b41b17a..b28f8f35a664 100644 --- a/iotdb-core/ainode/iotdb/ainode/core/model/chronos2/pipeline_chronos2.py +++ b/iotdb-core/ainode/iotdb/ainode/core/model/chronos2/pipeline_chronos2.py @@ -34,7 +34,7 @@ class Chronos2Pipeline(ForecastPipeline): def __init__(self, model_info, **model_kwargs): - super().__init__(model_info, model_kwargs=model_kwargs) + super().__init__(model_info, **model_kwargs) def preprocess(self, inputs, **infer_kwargs): """ diff --git a/iotdb-core/ainode/iotdb/ainode/core/model/sktime/pipeline_sktime.py b/iotdb-core/ainode/iotdb/ainode/core/model/sktime/pipeline_sktime.py index 964ab156e264..12b2668543ef 100644 --- a/iotdb-core/ainode/iotdb/ainode/core/model/sktime/pipeline_sktime.py +++ b/iotdb-core/ainode/iotdb/ainode/core/model/sktime/pipeline_sktime.py @@ -31,7 +31,7 @@ class SktimePipeline(ForecastPipeline): def __init__(self, model_info: ModelInfo, **model_kwargs): model_kwargs.pop("device", None) # sktime models run on CPU - super().__init__(model_info, model_kwargs=model_kwargs) + super().__init__(model_info, **model_kwargs) def preprocess( self, diff --git a/iotdb-core/ainode/iotdb/ainode/core/model/sundial/pipeline_sundial.py b/iotdb-core/ainode/iotdb/ainode/core/model/sundial/pipeline_sundial.py index 1715f190e32d..8aa9b175169c 100644 --- a/iotdb-core/ainode/iotdb/ainode/core/model/sundial/pipeline_sundial.py +++ b/iotdb-core/ainode/iotdb/ainode/core/model/sundial/pipeline_sundial.py @@ -28,7 +28,7 @@ class SundialPipeline(ForecastPipeline): def __init__(self, model_info: ModelInfo, **model_kwargs): - super().__init__(model_info, model_kwargs=model_kwargs) + super().__init__(model_info, **model_kwargs) def preprocess(self, inputs, **infer_kwargs) -> torch.Tensor: """ diff --git a/iotdb-core/ainode/iotdb/ainode/core/model/timer_xl/pipeline_timer.py b/iotdb-core/ainode/iotdb/ainode/core/model/timer_xl/pipeline_timer.py index bb54eed4ec6e..213e6102c8b6 100644 --- a/iotdb-core/ainode/iotdb/ainode/core/model/timer_xl/pipeline_timer.py +++ b/iotdb-core/ainode/iotdb/ainode/core/model/timer_xl/pipeline_timer.py @@ -28,7 +28,7 @@ class TimerPipeline(ForecastPipeline): def __init__(self, model_info: ModelInfo, **model_kwargs): - super().__init__(model_info, model_kwargs=model_kwargs) + super().__init__(model_info, **model_kwargs) def preprocess(self, inputs, **infer_kwargs) -> torch.Tensor: """