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: """