finish

Author: CRZbulabula

LICENSE

@@ -339,4 +339,14 @@ LMax Disruptor is open source software licensed under the Apache License 2.0 and
 Project page: https://github.com/LMAX-Exchange/disruptor
 License: https://github.com/LMAX-Exchange/disruptor/blob/master/LICENCE.txt
 
+--------------------------------------------------------------------------------
+
+The following files include code modified from LMax Disruptor project.
+
+./iotdb-core/ainode/iotdb/ainode/core/model/chronos2/*
+
+The chronos-forecasting is open source software licensed under the Apache License 2.0
+Project page: https://github.com/amazon-science/chronos-forecasting
+License: https://github.com/amazon-science/chronos-forecasting/blob/main/LICENSE
+
 --------------------------------------------------------------------------------

iotdb-core/ainode/iotdb/ainode/core/model/chronos2/pipeline_chronos2.py

@@ -388,4 +388,4 @@ def _predict_step(
         return prediction
 
     def postprocess(self, output: torch.Tensor):
-        return output[0].mean(dim=1)
+        return output[0].mean(dim=1, keepdim=True)

iotdb-core/ainode/pyproject.toml

@@ -21,7 +21,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "apache-iotdb-ainode"
-version = "2.0.6.dev"
+version = "2.0.7.dev"
 description = "Apache IoTDB AINode"
 readme = "README.md"
 authors = ["Apache Software Foundation <[email protected]>"]