This is an automated email from the ASF dual-hosted git repository.
yongzao pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/iotdb.git
The following commit(s) were added to refs/heads/master by this push:
new 3016bb54e48 [AINode] Add TimerXL to the built-in list (#15832)
3016bb54e48 is described below
commit 3016bb54e48294dbcc90fc47ec5d5cfedf0f64b5
Author: Yongzao <[email protected]>
AuthorDate: Thu Jun 26 20:22:15 2025 +0800
[AINode] Add TimerXL to the built-in list (#15832)
---
.../relational/it/schema/IoTDBDatabaseIT.java | 26 ----------------------
.../ainode/core/model/built_in_model_factory.py | 2 +-
.../iotdb/confignode/persistence/ModelInfo.java | 1 +
3 files changed, 2 insertions(+), 27 deletions(-)
diff --git
a/integration-test/src/test/java/org/apache/iotdb/relational/it/schema/IoTDBDatabaseIT.java
b/integration-test/src/test/java/org/apache/iotdb/relational/it/schema/IoTDBDatabaseIT.java
index e173099f45e..555f4a5edc8 100644
---
a/integration-test/src/test/java/org/apache/iotdb/relational/it/schema/IoTDBDatabaseIT.java
+++
b/integration-test/src/test/java/org/apache/iotdb/relational/it/schema/IoTDBDatabaseIT.java
@@ -569,19 +569,6 @@ public class IoTDBDatabaseIT {
Collections.singleton(
"IOTDB-THRIFT-SINK,Builtin,org.apache.iotdb.commons.pipe.agent.plugin.builtin.connector.iotdb.thrift.IoTDBThriftConnector,null,"));
- TestUtils.assertResultSetEqual(
- statement.executeQuery(
- "select model_id from information_schema.models where model_type
= 'BUILT_IN_FORECAST'"),
- "model_id,",
- new HashSet<>(
- Arrays.asList(
- "_STLForecaster,",
- "_NaiveForecaster,",
- "_sundial,",
- "_HoltWinters,",
- "_ExponentialSmoothing,",
- "_ARIMA,")));
-
TestUtils.assertResultSetEqual(
statement.executeQuery(
"select distinct(function_type) from
information_schema.functions"),
@@ -694,19 +681,6 @@ public class IoTDBDatabaseIT {
Collections.singleton(
"test,view_table,CREATE VIEW \"view_table\" (\"tag1\" STRING
TAG,\"tag2\" STRING TAG,\"s11\" INT32 FIELD,\"s3\" INT32 FIELD FROM \"s2\")
RESTRICT WITH (ttl=100) AS root.\"a\".**,"));
- TestUtils.assertResultSetEqual(
- statement.executeQuery(
- "select model_id from information_schema.models where model_type
= 'BUILT_IN_FORECAST'"),
- "model_id,",
- new HashSet<>(
- Arrays.asList(
- "_sundial,",
- "_STLForecaster,",
- "_NaiveForecaster,",
- "_HoltWinters,",
- "_ARIMA,",
- "_ExponentialSmoothing,")));
-
TestUtils.assertResultSetEqual(
statement.executeQuery(
"select distinct(function_type) from
information_schema.functions"),
diff --git a/iotdb-core/ainode/ainode/core/model/built_in_model_factory.py
b/iotdb-core/ainode/ainode/core/model/built_in_model_factory.py
index 9f72afdda91..806ff0ac204 100644
--- a/iotdb-core/ainode/ainode/core/model/built_in_model_factory.py
+++ b/iotdb-core/ainode/ainode/core/model/built_in_model_factory.py
@@ -50,7 +50,7 @@ def download_built_in_model_if_necessary(model_id: str,
local_dir):
"""
Download the built-in model from HuggingFace repository when necessary.
"""
- if "_timer" == model_id or "_sundial" == model_id:
+ if "timer" in model_id or "sundial" in model_id:
weights_path = os.path.join(local_dir, "model.safetensors")
if not os.path.exists(weights_path):
logger.info(
diff --git
a/iotdb-core/confignode/src/main/java/org/apache/iotdb/confignode/persistence/ModelInfo.java
b/iotdb-core/confignode/src/main/java/org/apache/iotdb/confignode/persistence/ModelInfo.java
index 0a5a7d235c1..fb9076626ab 100644
---
a/iotdb-core/confignode/src/main/java/org/apache/iotdb/confignode/persistence/ModelInfo.java
+++
b/iotdb-core/confignode/src/main/java/org/apache/iotdb/confignode/persistence/ModelInfo.java
@@ -80,6 +80,7 @@ public class ModelInfo implements SnapshotProcessor {
builtInForecastModel.add("_STLForecaster");
builtInForecastModel.add("_HoltWinters");
builtInForecastModel.add("_ExponentialSmoothing");
+ builtInForecastModel.add("_timerxl");
builtInForecastModel.add("_sundial");
builtInAnomalyDetectionModel.add("_GaussianHMM");
builtInAnomalyDetectionModel.add("_GMMHMM");
