fix ut

Author: zhiqiu

python/paddle/fluid/optimizer.py

@@ -42,6 +42,7 @@
 from functools import cmp_to_key
 from .wrapped_decorator import signature_safe_contextmanager
 from .. import compat as cpt
+import warnings
 
 __all__ = [
     'SGD', 'Momentum', 'Adagrad', 'Adam', 'Adamax', 'Dpsgd', 'DecayedAdagrad',
@@ -993,7 +994,7 @@ def flatten_param_grads(self, params_grads):
                     p, 'regularizer', None) is not None:
                 warnings.warn(
                     "flatten_param_grads=True will be discarded since paramter '{}''s need_clip is False or "
-                    "the regularizer is set for parameter".format(p.name))
+                    "the regularizer is set".format(p.name))
                 self._flatten_param_grads = False
                 return params_grads
 

python/paddle/fluid/tests/unittests/test_adam_op.py

@@ -785,6 +785,7 @@ def test_adam_api(self):
 
     def test_adam_flatten_param_grads_with_regularizer(self):
         # flatten_param_grads + regularizer is not supported yet.
+        paddle.enable_static()
         main = fluid.Program()
         weight_attr = paddle.ParamAttr(
             name="weight1",
@@ -805,6 +806,7 @@ def test_adam_flatten_param_grads_with_regularizer(self):
             adam = fluid.optimizer.AdamOptimizer(
                 0.01, flatten_param_grads=True, align_size=256)
             adam.minimize(avg_cost)
+            paddle.disable_static()
 
             self.assertEqual(adam._flatten_param_grads, False)