diff --git a/python/paddle/fluid/layers/control_flow.py b/python/paddle/fluid/layers/control_flow.py index 21454370dd28f02a767ad3935021c13f4205eabc..9d98e8333ba07ac3eed3a3b63adcba1919cb4694 100644 --- a/python/paddle/fluid/layers/control_flow.py +++ b/python/paddle/fluid/layers/control_flow.py @@ -235,8 +235,7 @@ class BlockGuardWithCompletion(BlockGuard): def __init__(self, rnn): if not isinstance(rnn, StaticRNN): - raise TypeError( - "BlockGuardWithCompletion takes a StaticRNN or ParallelDo") + raise TypeError("BlockGuardWithCompletion takes a StaticRNN") super(BlockGuardWithCompletion, self).__init__(rnn.helper.main_program) self.rnn = rnn