def METHOD_NAME():
    block = head_module.RegressionHead(dropout=0, shape=(8,))
    outputs = block.build(
        keras_tuner.HyperParameters(),
        keras.Input(shape=(5,), dtype=tf.float32),
    )
    assert len(nest.flatten(outputs)) == 1