From 16099abfffaf82f3af70d4fe5e9b2eda592b0125 Mon Sep 17 00:00:00 2001
From: zhiboniu <31800336+zhiboniu@users.noreply.github.com>
Date: Wed, 16 Jun 2021 17:32:12 +0800
Subject: [PATCH] fix new ci check errors (#33561)

---
 python/paddle/fluid/__init__.py                             | 6 ++++--
 python/paddle/fluid/contrib/__init__.py                     | 1 +
 python/paddle/fluid/contrib/mixed_precision/__init__.py     | 4 +++-
 .../paddle/fluid/contrib/mixed_precision/bf16/__init__.py   | 3 ++-
 4 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/python/paddle/fluid/__init__.py b/python/paddle/fluid/__init__.py
index 3b73034dfde..f9e0e0ae047 100644
--- a/python/paddle/fluid/__init__.py
+++ b/python/paddle/fluid/__init__.py
@@ -92,6 +92,10 @@ from .dygraph.checkpoint import save_dygraph, load_dygraph
 from .dygraph.varbase_patch_methods import monkey_patch_varbase
 from . import generator
 from .core import _cuda_synchronize
+from .generator import Generator
+from .trainer_desc import TrainerDesc, DistMultiTrainer, PipelineTrainer, MultiTrainer, HeterXpuTrainer, HeterBoxTrainer
+from .transpiler import HashName, RoundRobin
+from .backward import append_backward
 
 Tensor = LoDTensor
 enable_imperative = enable_dygraph
@@ -116,7 +120,6 @@ __all__ = framework.__all__ + executor.__all__ + \
         'transpiler',
         'nets',
         'optimizer',
-        'learning_rate_decay',
         'backward',
         'regularizer',
         'LoDTensor',
@@ -137,7 +140,6 @@ __all__ = framework.__all__ + executor.__all__ + \
         'install_check',
         'save',
         'load',
-        'VarBase',
         '_cuda_synchronize'
     ]
 
diff --git a/python/paddle/fluid/contrib/__init__.py b/python/paddle/fluid/contrib/__init__.py
index 0221a42e2a3..64e7eb395b7 100644
--- a/python/paddle/fluid/contrib/__init__.py
+++ b/python/paddle/fluid/contrib/__init__.py
@@ -33,6 +33,7 @@ from .mixed_precision import *
 from . import layers
 from .layers import *
 from . import optimizer
+from .optimizer import *
 from . import sparsity
 from .sparsity import *
 
diff --git a/python/paddle/fluid/contrib/mixed_precision/__init__.py b/python/paddle/fluid/contrib/mixed_precision/__init__.py
index a580ae5574c..1dd5015ec80 100644
--- a/python/paddle/fluid/contrib/mixed_precision/__init__.py
+++ b/python/paddle/fluid/contrib/mixed_precision/__init__.py
@@ -20,7 +20,9 @@ from . import fp16_lists
 from .fp16_lists import *
 from . import fp16_utils
 from .fp16_utils import *
+from . import bf16
 
-__all__ = decorator.__all__
+__all__ = []
+__all__ += decorator.__all__
 __all__ += fp16_lists.__all__
 __all__ += fp16_utils.__all__
diff --git a/python/paddle/fluid/contrib/mixed_precision/bf16/__init__.py b/python/paddle/fluid/contrib/mixed_precision/bf16/__init__.py
index d3632729a3b..0920176f772 100644
--- a/python/paddle/fluid/contrib/mixed_precision/bf16/__init__.py
+++ b/python/paddle/fluid/contrib/mixed_precision/bf16/__init__.py
@@ -21,6 +21,7 @@ from .amp_utils import *
 from . import decorator
 from .decorator import *
 
-__all__ = decorator.__all__
+__all__ = []
+__all__ += decorator.__all__
 __all__ += amp_lists.__all__
 __all__ += amp_utils.__all__
-- 
GitLab