code: def get_canonical_stage(stage):
      key = stage.lower()
      if key not in _CANONICAL_MAPPING:
          raise MlflowException(
              "Invalid Model Version stage: {}. Value must be one of {}.".format(
                  stage, ", ".join(ALL_STAGES)
              ),
              INVALID_PARAMETER_VALUE,
          )
      return _CANONICAL_MAPPING[key]
  
label_desc: Use image node_id 1 for calling a global function with example usage: get_canonical_stage(stage) and returns: _CANONICAL_MAPPING