code: def encode_map(features, labels):
      encoded_features = tuple(
          tf.expand_dims(features[key], axis=1)
          for key in sorted(features.keys())
      )
      return (encoded_features, labels)
  
imports: import pandas
  import tensorflow
  import wandb
label_desc: Use image node_id 1 for calling a global function with example usage: encode_map(features, labels) and returns: encoded_features,  labels