From 32d2d3b50f2aeea1cf17a24b295e29678d021966 Mon Sep 17 00:00:00 2001
From: yaoxuefeng6 <yaoxuefeng@baidu.com>
Date: Tue, 22 Sep 2020 09:51:11 +0800
Subject: [PATCH] fix ut

---
 python/paddle/fluid/tests/unittests/test_data_generator.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python/paddle/fluid/tests/unittests/test_data_generator.py b/python/paddle/fluid/tests/unittests/test_data_generator.py
index acff800a177..310913f695b 100644
--- 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):
-- 
GitLab