fix MultiSlotDataGenerator error (#36773)

dc0178ef · seemingwang · GitHub · 704e454f · dc0178ef
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/distributed/fleet/data_generator/__init__.py python/paddle/distributed/fleet/data_generator/__init__.py +1 -1

未找到文件。
--- a/python/paddle/distributed/fleet/data_generator/__init__.py
+++ b/python/paddle/distributed/fleet/data_generator/__init__.py
@@ -11,6 +11,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and

-from .data_generator import DataGenerator  # noqa: F401
+from .data_generator import DataGenerator, MultiSlotDataGenerator  # noqa: F401

 __all__ = []