This is an automated email from the ASF dual-hosted git repository.

yongzao pushed a commit to branch bug-fix-dataset
in repository https://gitbox.apache.org/repos/asf/iotdb.git

commit 3fd4853de5b50997ca664296206b0fdb7db044d8
Author: Yongzao <[email protected]>
AuthorDate: Thu Jul 3 18:27:45 2025 +0800

    Update iotdb.py
---
 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],

Reply via email to