fix UT

b70002c5 · seiriosPlus · ef2b1af4 · b70002c5
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

python/paddle/fluid/tests/unittests/test_dist_fleet_ps4.py python/paddle/fluid/tests/unittests/test_dist_fleet_ps4.py +2 -2

未找到文件。
--- a/python/paddle/fluid/tests/unittests/test_dist_fleet_ps4.py
+++ b/python/paddle/fluid/tests/unittests/test_dist_fleet_ps4.py
@@ -68,7 +68,7 @@ class TestPSPassWithBow(unittest.TestCase):
            name="query_ids", shape=[1], dtype="int64", lod_level=1)
        # embedding
        q_emb = fluid.contrib.layers.sparse_embedding(
-            input=pt,
+            input=q,
            size=[dict_dim, emb_dim],
            param_attr=fluid.ParamAttr(
                initializer=fluid.initializer.Constant(value=0.01),
@@ -117,7 +117,7 @@ class TestPSPassWithBow(unittest.TestCase):
            name="neg_title_ids", shape=[1], dtype="int64", lod_level=1)
        # embedding
        nt_emb = fluid.contrib.layers.sparse_embedding(
-            input=pt,
+            input=nt,
            size=[dict_dim, emb_dim],
            param_attr=fluid.ParamAttr(
                initializer=fluid.initializer.Constant(value=0.01),