add int cast, test=develop

aed3872c · tangwei12 · 41790f13 · aed3872c
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

python/paddle/fluid/tests/unittests/test_nce_remote_table_op.py .../paddle/fluid/tests/unittests/test_nce_remote_table_op.py +1 -0

未找到文件。
--- 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,