saving a TFRecord dataset

Question

I have converted a dataframe to a TFRecord dataset as the following:

import pandas as pd
import tensorflow as tf
from pandas_tfrecords import pd2tf, tf2pd

df = p