add int cast, test=develop

6 years ago · aed3872c1c
parent 41790f1366
commit aed3872c1c
1 changed files with 1 additions and 0 deletions
--- 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,