d97e9cd0c30980647f31c1003d6367e6a41c7124,tests/models/DCN_test.py,,test_DCN_invalid,#Any#Any#Any#,23

Before Change

def test_DCN_invalid(embedding_size=8, cross_num=0, hidden_size=()):
    feature_dim_dict = {"sparse": {"sparse_1": 2, "sparse_2": 5,
                                   "sparse_3": 10}, "dense": ["dense_1", "dense_2", "dense_3"]}
    with pytest.raises(ValueError):
        _ = DCN(feature_dim_dict, embedding_size=embedding_size, cross_num=cross_num,
                hidden_size=hidden_size, keep_prob=0.5, )

After Change

def test_DCN_invalid(embedding_size=8, cross_num=0, hidden_size=()):
    feature_dim_dict = {"sparse": [SingleFeat("sparse_1",2),SingleFeat("sparse_2",5),SingleFeat("sparse_3",10)],
    "dense": [SingleFeat("dense_1",1),SingleFeat("dense_1",1),SingleFeat("dense_1",1)]}
    with pytest.raises(ValueError):
        _ = DCN(feature_dim_dict, embedding_size=embedding_size, cross_num=cross_num,

Instances