From 77ed03d6a760dbb6173b0ca5dec92ab03df06016 Mon Sep 17 00:00:00 2001
From: JYChen <zoooo0820@qq.com>
Date: Tue, 13 Dec 2022 19:10:30 +0800
Subject: [PATCH] remove non-public apis from __all__ (#48952)

* remove non-public apis from __all__

* fix code style
---
 python/paddle/jit/__init__.py                     |  3 +--
 python/paddle/jit/api.py                          | 13 +------------
 python/paddle/jit/dy2static/ifelse_transformer.py |  2 +-
 python/paddle/static/__init__.py                  |  2 --
 4 files changed, 3 insertions(+), 17 deletions(-)

diff --git a/python/paddle/jit/__init__.py b/python/paddle/jit/__init__.py
index af4aad939a..35d1be0aaf 100644
--- a/python/paddle/jit/__init__.py
+++ b/python/paddle/jit/__init__.py
@@ -15,11 +15,10 @@
 
 from .api import save
 from .api import load
-from .api import set_code_level
-from .api import set_verbosity
 from .api import declarative as to_static
 from .api import not_to_static
 from .api import TranslatedLayer
+from .dy2static.logging_utils import set_code_level, set_verbosity
 
 from . import dy2static
 from .dy2static.program_translator import ProgramTranslator
diff --git a/python/paddle/jit/api.py b/python/paddle/jit/api.py
index 4cadf127f3..fff8e37e60 100644
--- a/python/paddle/jit/api.py
+++ b/python/paddle/jit/api.py
@@ -42,10 +42,6 @@ from .dy2static.convert_call_func import (
     ConversionOptions,
     CONVERSION_OPTIONS,
 )
-from .dy2static.logging_utils import (
-    set_code_level,
-    set_verbosity,
-)
 from .dy2static.program_translator import (
     ProgramTranslator,
     StaticFunction,
@@ -76,14 +72,7 @@ from paddle.fluid.framework import (
 from paddle.fluid.framework import dygraph_only, _non_static_mode
 from paddle.fluid.wrapped_decorator import wrap_decorator
 
-__all__ = [
-    'declarative',
-    'set_code_level',
-    'set_verbosity',
-    'save',
-    'load',
-    'not_to_static',
-]
+__all__ = []
 
 
 def create_program_from_desc(program_desc):
diff --git a/python/paddle/jit/dy2static/ifelse_transformer.py b/python/paddle/jit/dy2static/ifelse_transformer.py
index 19d856310b..d6744bc65c 100644
--- a/python/paddle/jit/dy2static/ifelse_transformer.py
+++ b/python/paddle/jit/dy2static/ifelse_transformer.py
@@ -44,7 +44,7 @@ from paddle.utils import gast
 
 from .base_transformer import BaseTransformer
 
-__all__ = ['IfElseTransformer']
+__all__ = []
 
 TRUE_FUNC_PREFIX = 'true_fn'
 FALSE_FUNC_PREFIX = 'false_fn'
diff --git a/python/paddle/static/__init__.py b/python/paddle/static/__init__.py
index bdb88e7290..292a1da119 100644
--- a/python/paddle/static/__init__.py
+++ b/python/paddle/static/__init__.py
@@ -74,13 +74,11 @@ from ..fluid.io import batch  # noqa: F401
 from ..fluid.contrib.layers import ctr_metric_bundle  # noqa: F401
 from ..fluid.layers import exponential_decay  # noqa: F401
 
-from .nn.common import batch_norm  # noqa: F401
 from .nn.metric import auc  # noqa: F401
 from .nn.metric import accuracy  # noqa: F401
 
 __all__ = [  # noqa
     'append_backward',
-    'batch_norm',
     'gradients',
     'Executor',
     'global_scope',
-- 
GitLab