[AINode] IoTDBTreeModelDataset bug fix (#15869)

CRZbulabula · web-flow · commit baa9e1ac0050 · 2025-07-03T20:16:37.000+08:00
diff --git a/iotdb-core/ainode/ainode/core/ingress/iotdb.py b/iotdb-core/ainode/ainode/core/ingress/iotdb.py
@@ -223,6 +223,7 @@ def __getitem__(self, index):
             + 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],

Original file line number	Diff line number	Diff line change
`@@ -223,6 +223,7 @@ def __getitem__(self, index):`
`223`	`223`	`+ self.seq_len`
`224`	`224`	`+ self.output_token_len`
`225`	`225`	`]`
	`226`	`+ result = torch.tensor(result)`
`226`	`227`	`return (`
`227`	`228`	`result[0 : self.seq_len],`
`228`	`229`	`result[self.input_token_len : self.seq_len + self.output_token_len],`