«KeyError: 'partitioner'» выдается, когда Tensorflow 2.2 загружает модель 2.1 с NumericColumn.

Question

Я построил функциональную модель в Tensorflow 2.1, которая использует NumericColumn и сохранила ее в формате h5.

Она хорошо загружается в 2.1, но в 2.2 выдает эту ошибку:

Traceback (most recent call last):
  File "my_code.py", line 75, in _load_model_from_path
    model = tf.keras.models.load_model(model_path)
  File ".venv/lib/python3.6/site-packages/tensorflow/python/keras/saving/save.py", line 184, in load_model
    return hdf5_format.load_model_from_hdf5(filepath, custom_objects, compile)
  File ".venv/lib/python3.6/site-packages/tensorflow/python/keras/saving/hdf5_format.py", line 178, in load_model_from_hdf5
    custom_objects=custom_objects)
  File ".venv/lib/python3.6/site-packages/tensorflow/python/keras/saving/model_config.py", line 55, in model_from_config
    return deserialize(config, custom_objects=custom_objects)
  File ".venv/lib/python3.6/site-packages/tensorflow/python/keras/layers/serialization.py", line 109, in deserialize
    printable_module_name='layer')
  File ".venv/lib/python3.6/site-packages/tensorflow/python/keras/utils/generic_utils.py", line 373, in deserialize_keras_object
    list(custom_objects.items())))
  File ".venv/lib/python3.6/site-packages/tensorflow/python/keras/engine/network.py", line 987, in from_config
    config, custom_objects)
  File ".venv/lib/python3.6/site-packages/tensorflow/python/keras/engine/network.py", line 2019, in reconstruct_from_config
    process_layer(layer_data)
  File ".venv/lib/python3.6/site-packages/tensorflow/python/keras/engine/network.py", line 2001, in process_layer
    layer = deserialize_layer(layer_data, custom_objects=custom_objects)
  File ".venv/lib/python3.6/site-packages/tensorflow/python/keras/layers/serialization.py", line 109, in deserialize
    printable_module_name='layer')
  File ".venv/lib/python3.6/site-packages/tensorflow/python/keras/utils/generic_utils.py", line 373, in deserialize_keras_object
    list(custom_objects.items())))
  File ".venv/lib/python3.6/site-packages/tensorflow/python/feature_column/feature_column_v2.py", line 473, in from_config
    config['partitioner'], custom_objects)
KeyError: 'partitioner'

Неизвестный ключ partitioner ожидается в config слоя, поэтому, похоже, это проблема обратной несовместимости.

Можно ли обойти это?