Issue with generating examples for a loaded tf.estimator when using tf.function

Question

I\'m using tensorflow 2.4 and trying to generate examples for a loaded estimator (tf.estimator.BoostedTreesRegressor).

I\'m using tf.function() to be able to save the