fix test_understand_sentiment bug (#8767)

7 years ago · bc36396d71
parent d398b96823
commit bc36396d71
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/fluid/tests/book/test_understand_sentiment.py
+++ b/python/paddle/fluid/tests/book/test_understand_sentiment.py
@ -274,7 +274,7 @@ def main(word_dict, net_method, use_cuda, parallel=False, save_dirname=None):
        use_cuda,
        parallel=parallel,
        save_dirname=save_dirname)
-    infer(use_cuda, save_dirname)
+    infer(word_dict, use_cuda, save_dirname)


 class TestUnderstandSentiment(unittest.TestCase):