3a431ea52d090fb3ef8a1e0e5d7f796d9a42e097,tests/keras/applications/applications_test.py,,test_densenet_pooling,#Any#Any#,357

Before Change

    p = Process(target=target, args=(queue,))
    p.start()
    p.join()
    assert not queue.empty(), "Model creation failed."
    model_output_shape = queue.get_nowait()
    assert model_output_shape == (None, None, None, dim)

After Change

def test_densenet_pooling(fun, dim):
    def model_fn():
        return fun(weights=None, include_top=False, pooling="avg")
    output_shape = clean_run(model_fn)
    assert output_shape == (None, None, None, dim)


@keras_test

Instances