fix unittest

PaddlePaddle · Sep 25, 2023 · 8af28f6 · 8af28f6
1 parent ddd8fc4
commit 8af28f6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/legacy_test/test_lr_scheduler.py b/test/legacy_test/test_lr_scheduler.py
@@ -987,7 +987,7 @@ def test_scheduler(self):
                 paddle.optimizer.lr.LinearLR,
                 {
                     "learning_rate": 0.2,
-                    "total_steps:": 40,
+                    "total_steps": 40,
                     "start_factor": 0.5,
                     "end_factor": 1,
                     "verbose": False,
@@ -998,7 +998,7 @@ def test_scheduler(self):
                 paddle.optimizer.lr.LinearLR,
                 {
                     "learning_rate": 0.2,
-                    "total_steps:": 5,
+                    "total_steps": 5,
                     "start_factor": 0.2,
                     "end_factor": 0.5,
                     "verbose": False,