From aed3872c1c5c0c9957f9567071f63a89c1ace455 Mon Sep 17 00:00:00 2001
From: tangwei12 <tangwei12@baidu.com>
Date: Tue, 18 Dec 2018 16:17:20 +0800
Subject: [PATCH] add int cast, test=develop

---
 python/paddle/fluid/tests/unittests/test_nce_remote_table_op.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python/paddle/fluid/tests/unittests/test_nce_remote_table_op.py b/python/paddle/fluid/tests/unittests/test_nce_remote_table_op.py
index e87545cb9..5e440bf35 100644
--- a/python/paddle/fluid/tests/unittests/test_nce_remote_table_op.py
+++ b/python/paddle/fluid/tests/unittests/test_nce_remote_table_op.py
@@ -141,6 +141,7 @@ class TestListenAndServOp(unittest.TestCase):
                     SampleLabels='SampleLabels',
                     num_total_classes=5,
                     num_neg_samples=2,
+                    custom_neg_classes=list(range(2)),
                     sampler=0,
                     seed=1,
                     is_sparse=True,
-- 
GitLab