fix ut

32d2d3b5 · yaoxuefeng6 · 51178908 · 32d2d3b5
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

python/paddle/fluid/tests/unittests/test_data_generator.py python/paddle/fluid/tests/unittests/test_data_generator.py +1 -0

未找到文件。
--- a/python/paddle/fluid/tests/unittests/test_data_generator.py
+++ b/python/paddle/fluid/tests/unittests/test_data_generator.py
@@ -14,6 +14,7 @@ import paddle
 import unittest
 import paddle.distributed.fleet as fleet
 import os
+import sys
 class MyMultiSlotDataGenerator(fleet.MultiSlotDataGenerator):