Merge branch 'beta' into release/v1.1.0

duyifanict · duyifanict · commit a360ac10e455 · 2025-12-19T11:45:29.000+08:00
diff --git a/tests/smoke_test/test_koopa.py b/tests/smoke_test/test_koopa.py
@@ -8,7 +8,7 @@
 from basicts.configs import BasicTSForecastingConfig
 from basicts.launcher import BasicTSLauncher
 from basicts.models.Koopa import Koopa, KoopaConfig
-from basicts.runners.callback import KoopaMaskInitCallbackFullTrain
+from basicts.models.Koopa.callback.koopa_mask_init import KoopaMaskInitCallback
 
 
 def test_koopa_smoke_test():
@@ -33,7 +33,7 @@ def test_koopa_smoke_test():
             input_len=input_len,
             output_len=output_len,
             lr=0.001,
-            callbacks=[KoopaMaskInitCallbackFullTrain(alpha=0.2)],
+            callbacks=[KoopaMaskInitCallback(alpha=0.2)],
         )
     )
 

Original file line number	Diff line number	Diff line change
`@@ -8,7 +8,7 @@`
`8`	`8`	`from basicts.configs import BasicTSForecastingConfig`
`9`	`9`	`from basicts.launcher import BasicTSLauncher`
`10`	`10`	`from basicts.models.Koopa import Koopa, KoopaConfig`
`11`		`-from basicts.runners.callback import KoopaMaskInitCallbackFullTrain`
	`11`	`+from basicts.models.Koopa.callback.koopa_mask_init import KoopaMaskInitCallback`
`12`	`12`
`13`	`13`
`14`	`14`	`def test_koopa_smoke_test():`
`@@ -33,7 +33,7 @@ def test_koopa_smoke_test():`
`33`	`33`	`input_len=input_len,`
`34`	`34`	`output_len=output_len,`
`35`	`35`	`lr=0.001,`
`36`		`- callbacks=[KoopaMaskInitCallbackFullTrain(alpha=0.2)],`
	`36`	`+ callbacks=[KoopaMaskInitCallback(alpha=0.2)],`
`37`	`37`	`)`
`38`	`38`	`)`
`39`	`39`