code: def create_context_type(
      store: metadata_store.MetadataStore, typename: str
  ) -> metadata_store_pb2.ContextType:
      """Put a Context Type in the MLMD database."""
      context_type = metadata_store_pb2.ContextType(name=typename)
      context_type.id = store.put_context_type(context_type)
      return context_type
  
imports: from typing import Dict, List
  from absl.testing import absltest
  from ml_metadata import metadata_store
  from ml_metadata.proto import metadata_store_pb2
  from ml_metadata.tools.mlmd_resolver import metadata_resolver
  from ml_metadata.tools.mlmd_resolver import metadata_resolver_utils
label_desc: Use image node_id 5 for calling a global function with example usage: create_context_type(store, typename) and returns: context_type