Revert "fix _message of protobuf (#49673)"

This reverts commit 7fe44fe.
PaddlePaddle · Mar 2, 2023 · 2827b59 · 2827b59
1 parent 5e1185d
commit 2827b59
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/python/paddle/distributed/fleet/base/distributed_strategy.py b/python/paddle/distributed/fleet/base/distributed_strategy.py
@@ -22,12 +22,6 @@
 from paddle.fluid.framework import _global_flags
 from paddle.fluid.wrapped_decorator import wrap_decorator
 
-protobuf_version = google.protobuf.__version__
-if protobuf_version >= "4.21.0":
-    import google._upb._message as _message
-else:
-    import google.protobuf.pyext._message as _message
-
 __all__ = []
 
 non_auto_func_called = True
@@ -2503,7 +2497,7 @@ def __repr__(self):
                             for ff in config_fields:
                                 if isinstance(
                                     getattr(my_configs, ff.name),
-                                    _message.RepeatedScalarContainer,
+                                    google.protobuf.pyext._message.RepeatedScalarContainer,
                                 ):
                                     values = getattr(my_configs, ff.name)
                                     for i, v in enumerate(values):