From a37be82fddff46235267961c462b48451d3b729e Mon Sep 17 00:00:00 2001
From: Sing_chan <51314274+betterpig@users.noreply.github.com>
Date: Thu, 16 Dec 2021 22:05:01 +0800
Subject: [PATCH] modify according to zhouwei's comment (#38166)

---
 .../paddle/fluid/tests/unittests/test_imperative_optimizer.py  | 3 ++-
 .../fluid/tests/unittests/test_imperative_optimizer_v2.py      | 3 ++-
 tools/windows/run_unittests.sh                                 | 2 --
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/python/paddle/fluid/tests/unittests/test_imperative_optimizer.py b/python/paddle/fluid/tests/unittests/test_imperative_optimizer.py
index 36c4d67bf2d..0b1f5c91119 100644
--- a/python/paddle/fluid/tests/unittests/test_imperative_optimizer.py
+++ b/python/paddle/fluid/tests/unittests/test_imperative_optimizer.py
@@ -632,7 +632,8 @@ class TestImperativeLambOptimizer(TestImperativeOptimizerBase):
             learning_rate=0.002, exclude_from_weight_decay_fn=exclude_fn)
         return optimizer
 
-    def test_lamb(self):
+    # should fix: may fail in CI-windows
+    def _test_lamb(self):
         self._check_mlp()
 
 
diff --git a/python/paddle/fluid/tests/unittests/test_imperative_optimizer_v2.py b/python/paddle/fluid/tests/unittests/test_imperative_optimizer_v2.py
index eac627d1b5b..dfd1e4f97a8 100644
--- a/python/paddle/fluid/tests/unittests/test_imperative_optimizer_v2.py
+++ b/python/paddle/fluid/tests/unittests/test_imperative_optimizer_v2.py
@@ -721,7 +721,8 @@ class TestImperativeLambOptimizer(TestImperativeOptimizerBase):
             learning_rate=0.002, exclude_from_weight_decay_fn=exclude_fn)
         return optimizer
 
-    def test_lamb(self):
+    # should fix: may fail in CI-windows
+    def _test_lamb(self):
         self._check_mlp()
 
 
diff --git a/tools/windows/run_unittests.sh b/tools/windows/run_unittests.sh
index b4c8030a1ee..24005b7336c 100644
--- a/tools/windows/run_unittests.sh
+++ b/tools/windows/run_unittests.sh
@@ -95,7 +95,6 @@ disable_wingpu_test="^test_model$|\
 ^test_activation_op$|\
 ^test_norm_nn_grad$|\
 ^test_bilinear_interp_op$|\
-^test_imperative_optimizer_v2$|\
 ^disable_wingpu_test$"
 
 # /*==================Fixed Disabled Windows GPU MKL unittests==============================*/
@@ -188,7 +187,6 @@ long_time_test="^test_gru_op$|\
 ^test_sgd_op$|\
 ^test_transformer$|\
 ^test_imperative_auto_mixed_precision$|\
-^test_imperative_optimizer_v2$|\
 ^test_trt_matmul_quant_dequant$|\
 ^test_strided_slice_op$"
 
-- 
GitLab