I\'m following this tensorflow tutorial https://github.com/tensorflow/hub/blob/master/examples/colab/tf2_semantic_approximate_nearest_neighbors.ipynb When writing to tfRecor