fs_wrapper add __all__ (#24335)

python/paddle/__init__.py

@@ -30,9 +30,9 @@ except ImportError:
 import paddle.reader
 import paddle.dataset
 import paddle.batch
+batch = batch.batch
 import paddle.compat
 import paddle.distributed
-batch = batch.batch
 import paddle.sysconfig
 import paddle.tensor
 import paddle.nn

python/paddle/distributed/__init__.py

@@ -11,3 +11,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+from . import fs_wrapper
+
+__all__ = fs_wrapper.__all__

python/paddle/distributed/fs_wrapper.py

@@ -19,6 +19,8 @@ import os
 from pathlib import PurePosixPath
 import shutil
 
+__all__ = ['FS', 'LocalFS', 'BDFS']
+
 
 class FS(object):
     @abc.abstractmethod