polish code for reading. test=develop

6 years ago · 6deac40724
parent d0a2a202d0
commit 6deac40724
1 changed files with 1 additions and 0 deletions
--- a/python/paddle/fluid/tests/unittests/test_ir_memory_optimize_transformer.py
+++ b/python/paddle/fluid/tests/unittests/test_ir_memory_optimize_transformer.py
@ -22,6 +22,7 @@ os.environ[
    'RECORDIO_FILENAME'] = '/tmp/ir_memory_optimize_transformer.wmt16.recordio'

 from test_parallel_executor_transformer import TestTransformer
+from test_parallel_executor_transformer import transformer


 # NOTE(dzhwinter): test diferent strategy colisions.