def test_bert_build_return_tensor():
    block = blocks.BertBlock()

    outputs = block.build(
        keras_tuner.HyperParameters(), keras.Input(shape=(1,), dtype=tf.string)