This is an automated email from the ASF dual-hosted git repository. ycycse pushed a commit to branch inferenceFix in repository https://gitbox.apache.org/repos/asf/iotdb.git
commit 5a6c9bc848d651cc60f9873453645a79e3bec813 Author: YangCaiyin <[email protected]> AuthorDate: Thu May 15 14:38:42 2025 +0800 fix input error --- iotdb-core/ainode/ainode/core/manager/inference_manager.py | 4 +--- .../plan/relational/function/tvf/ForecastTableFunction.java | 2 +- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/iotdb-core/ainode/ainode/core/manager/inference_manager.py b/iotdb-core/ainode/ainode/core/manager/inference_manager.py index c9815745516..e5142b0205b 100644 --- a/iotdb-core/ainode/ainode/core/manager/inference_manager.py +++ b/iotdb-core/ainode/ainode/core/manager/inference_manager.py @@ -55,9 +55,7 @@ class TimerXLStrategy(InferenceStrategy): class BuiltInStrategy(InferenceStrategy): def infer(self, full_data, **_): - data = full_data[1][0] - if data.dtype.byteorder not in ('=', '|'): - data = data.byteswap().newbyteorder() + data = pd.DataFrame(full_data[1]).T output = self.model.inference(data) df = pd.DataFrame(output) return convert_to_binary(df) diff --git a/iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/plan/relational/function/tvf/ForecastTableFunction.java b/iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/plan/relational/function/tvf/ForecastTableFunction.java index 9086e6f5edd..35dbed2ac9f 100644 --- a/iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/plan/relational/function/tvf/ForecastTableFunction.java +++ b/iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/plan/relational/function/tvf/ForecastTableFunction.java @@ -503,7 +503,7 @@ public class ForecastTableFunction implements TableFunction { long inputEndTime = inputRecords.getLast().getLong(0); long interval = outputInterval <= 0 - ? (inputEndTime - inputStartTime) / inputRecords.size() + ? (inputEndTime - inputStartTime) / (inputRecords.size() - 1) : outputInterval; long outputTime = (outputStartTime == Long.MIN_VALUE) ? (inputEndTime + interval) : outputStartTime;
