AssertionError: Could not compute output Tensor(“keras_layer/cond/Identity:0”, shape=(None, 768), dtype=float32)

Question

I am following a tutorial to perform sentence embedding using BERT. Here the code:

import tensorflow as tf
import tensorflow_hub as hub
import bert
from tenso