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 baa9e1ac005 [AINode] IoTDBTreeModelDataset bug fix (#15869)
baa9e1ac005 is described below
commit baa9e1ac005047368312f3106392b8e6f2b24956
Author: Yongzao <[email protected]>
AuthorDate: Thu Jul 3 20:16:37 2025 +0800
[AINode] IoTDBTreeModelDataset bug fix (#15869)
---
iotdb-core/ainode/ainode/core/ingress/iotdb.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/iotdb-core/ainode/ainode/core/ingress/iotdb.py
b/iotdb-core/ainode/ainode/core/ingress/iotdb.py
index 528c3cb7397..2d1bb74016f 100644
--- a/iotdb-core/ainode/ainode/core/ingress/iotdb.py
+++ b/iotdb-core/ainode/ainode/core/ingress/iotdb.py
@@ -223,6 +223,7 @@ class IoTDBTreeModelDataset(BasicDatabaseForecastDataset):
+ self.seq_len
+ self.output_token_len
]
+ result = torch.tensor(result)
return (
result[0 : self.seq_len],
result[self.input_token_len : self.seq_len +
self.output_token_len],
