fix error

7 years ago · b00fbd962c
parent d684b49cb0
commit b00fbd962c
1 changed files with 2 additions and 2 deletions
--- a/python/paddle/fluid/layers/io.py
+++ b/python/paddle/fluid/layers/io.py
@ -637,10 +637,10 @@ def read_file(reader):
    out = [
        helper.create_tmp_variable(
            stop_gradient=True, dtype='float32')
-        for _ in range(len(file_obj.desc.shapes()))
+        for _ in range(len(reader.desc.shapes()))
    ]
    helper.append_op(
-        type='read', inputs={'Reader': [file_obj]}, outputs={'Out': out})
+        type='read', inputs={'Reader': [reader]}, outputs={'Out': out})
    if len(out) == 1:
        return out[0]
    else: