Fix the input of executor

7 years ago · fae5c1f514
parent 5b452bfd59
commit fae5c1f514
1 changed files with 3 additions and 2 deletions
--- a/python/paddle/fluid/executor.py
+++ b/python/paddle/fluid/executor.py
@ -320,8 +320,9 @@ class Executor(object):
        # append fetch_operators
        if not has_fetch_operators(global_block, fetch_list, fetch_var_name):
            for i, var in enumerate(fetch_list):
-                assert isinstance(var, Variable) or isinstance(var, str), (
-                    "Wrong type for fetch_list[%s]: %s" % (i, type(var)))
+                assert isinstance(var, Variable) or isinstance(
+                    var, six.string_types), (
+                        "Wrong type for fetch_list[%s]: %s" % (i, type(var)))
                global_block.append_op(
                    type='fetch',
                    inputs={'X': [var]},