Error loading model for AI Platform

MODEL_DIR: /data/bbhandari/aces-2/output/experiments_paro_before_during/DNN/trial_dnn_2023_08_22_v2

Traceback (most recent call last):
  File "/home/bbhandari/aces/workflow/v2/host_vertex_ai.py", line 24, in <module>
    this_model = tf.keras.models.load_model(f"{str(Config.MODEL_DIR)}/{Config.MODEL_NAME}.tf")
  File "/home/kmarkert/miniconda3/envs/cbias/lib/python3.9/site-packages/keras/utils/traceback_utils.py", line 67, in error_handler
    raise e.with_traceback(filtered_tb) from None
  File "/home/kmarkert/miniconda3/envs/cbias/lib/python3.9/site-packages/tensorflow/python/util/nest.py", line 570, in assert_same_structure
    raise type(e)("%s\n"
ValueError: The two structures don't have the same nested structure.

First structure: type=tuple str=(({'green_before': TensorSpec(shape=(None, 1, 1, 1), dtype=tf.float32, name=None), 'nir_before': TensorSpec(shape=(None, 1, 1, 1), dtype=tf.float32, name=None), 'green_during': TensorSpec(shape=(None, 1, 1, 1), dtype=tf.float32, name=None), 'nir_during': TensorSpec(shape=(None, 1, 1, 1), dtype=tf.float32, name=None), 'red_during': TensorSpec(shape=(None, 1, 1, 1), dtype=tf.float32, name=None), 'blue_before': TensorSpec(shape=(None, 1, 1, 1), dtype=tf.float32, name=None), 'blue_during': TensorSpec(shape=(None, 1, 1, 1), dtype=tf.float32, name=None), 'red_before': TensorSpec(shape=(None, 1, 1, 1), dtype=tf.float32, name=None)}, False), {})

Second structure: type=tuple str=(({'red_during': TensorSpec(shape=<unknown>, dtype=tf.float32, name='features_dict/red_during'), 'ndwi_during': TensorSpec(shape=<unknown>, dtype=tf.float32, name='features_dict/ndwi_during'), 'savi_before': TensorSpec(shape=<unknown>, dtype=tf.float32, name='features_dict/savi_before'), 'blue_before': TensorSpec(shape=<unknown>, dtype=tf.float32, name='features_dict/blue_before'), 'green_during': TensorSpec(shape=<unknown>, dtype=tf.float32, name='features_dict/green_during'), 'nir_before': TensorSpec(shape=<unknown>, dtype=tf.float32, name='features_dict/nir_before'), 'tgi_before': TensorSpec(shape=<unknown>, dtype=tf.float32, name='features_dict/tgi_before'), 'mtvi2_before': TensorSpec(shape=<unknown>, dtype=tf.float32, name='features_dict/mtvi2_before'), 'evi_during': TensorSpec(shape=<unknown>, dtype=tf.float32, name='features_dict/evi_during'), 'msavi_before': TensorSpec(shape=<unknown>, dtype=tf.float32, name='features_dict/msavi_before'), 'vari_during': TensorSpec(shape=<unknown>, dtype=tf.float32, name='features_dict/vari_during'), 'mtvi2_during': TensorSpec(shape=<unknown>, dtype=tf.float32, name='features_dict/mtvi2_during'), 'nir_during': TensorSpec(shape=<unknown>, dtype=tf.float32, name='features_dict/nir_during'), 'ndwi_before': TensorSpec(shape=<unknown>, dtype=tf.float32, name='features_dict/ndwi_before'), 'ndvi_during': TensorSpec(shape=<unknown>, dtype=tf.float32, name='features_dict/ndvi_during'), 'tgi_during': TensorSpec(shape=<unknown>, dtype=tf.float32, name='features_dict/tgi_during'), 'ndvi_before': TensorSpec(shape=<unknown>, dtype=tf.float32, name='features_dict/ndvi_before'), 'red_before': TensorSpec(shape=<unknown>, dtype=tf.float32, name='features_dict/red_before'), 'green_before': TensorSpec(shape=<unknown>, dtype=tf.float32, name='features_dict/green_before'), 'blue_during': TensorSpec(shape=<unknown>, dtype=tf.float32, name='features_dict/blue_during'), 'vari_before': TensorSpec(shape=<unknown>, dtype=tf.float32, name='features_dict/vari_before'), 'savi_during': TensorSpec(shape=<unknown>, dtype=tf.float32, name='features_dict/savi_during'), 'evi_before': TensorSpec(shape=<unknown>, dtype=tf.float32, name='features_dict/evi_before'), 'msavi_during': TensorSpec(shape=<unknown>, dtype=tf.float32, name='features_dict/msavi_during')}, False), {})

More specifically: The two dictionaries don't have the same set of keys. First structure has keys type=list str=['green_before', 'nir_before', 'green_during', 'nir_during', 'red_during', 'blue_before', 'blue_during', 'red_before'], while second structure has keys type=list str=['red_during', 'ndwi_during', 'savi_before', 'blue_before', 'green_during', 'nir_before', 'tgi_before', 'mtvi2_before', 'evi_during', 'msavi_before', 'vari_during', 'mtvi2_during', 'nir_during', 'ndwi_before', 'ndvi_during', 'tgi_during', 'ndvi_before', 'red_before', 'green_before', 'blue_during', 'vari_before', 'savi_during', 'evi_before', 'msavi_during']
Entire first structure:
(({'green_before': ., 'nir_before': ., 'green_during': ., 'nir_during': ., 'red_during': ., 'blue_before': ., 'blue_during': ., 'red_before': .}, .), {})
Entire second structure:
(({'red_during': ., 'ndwi_during': ., 'savi_before': ., 'blue_before': ., 'green_during': ., 'nir_before': ., 'tgi_before': ., 'mtvi2_before': ., 'evi_during': ., 'msavi_before': ., 'vari_during': ., 'mtvi2_during': ., 'nir_during': ., 'ndwi_before': ., 'ndvi_during': ., 'tgi_during': ., 'ndvi_before': ., 'red_before': ., 'green_before': ., 'blue_during': ., 'vari_before': ., 'savi_during': ., 'evi_before': ., 'msavi_during': .}, .), {})