code: def get_model_schema_and_features(model_dir):
    
    schema_file = os.path.join(model_dir, 'assets.extra', 'schema.json')
    schema = json.loads(file_io.read_file_to_string(schema_file))
    features_file = os.path.join(model_dir, 'assets.extra', 'features.json')
    features_config = json.loads(file_io.read_file_to_string(features_file))
    return schema, features_config
docs: Get a local model's schema and features config.
  
    Args:
      model_dir: local or GCS path of a model.
    Returns:
      A tuple of schema (list) and features config (dict).