def refactor_model_blessing(model_blessing: artifact.Artifact, name_from_id: Mapping[int, str]) -> None: """Changes id-typed custom properties to string-typed runtime artifact name.""" if model_blessing.has_custom_property( constants.ARTIFACT_PROPERTY_BASELINE_MODEL_ID_KEY): model_blessing.set_string_custom_property( constants.ARTIFACT_PROPERTY_BASELINE_MODEL_ID_KEY, _get_full_name(artifact_id=model_blessing.get_int_custom_property( constants.ARTIFACT_PROPERTY_BASELINE_MODEL_ID_KEY), name_from_id=name_from_id)) if model_blessing.has_custom_property( constants.ARTIFACT_PROPERTY_CURRENT_MODEL_ID_KEY): model_blessing.set_string_custom_property( constants.ARTIFACT_PROPERTY_CURRENT_MODEL_ID_KEY, _get_full_name(artifact_id=model_blessing.get_int_custom_property( constants.ARTIFACT_PROPERTY_CURRENT_MODEL_ID_KEY), name_from_id=name_from_id))
def _get_data_view_info( examples: artifact.Artifact) -> Optional[Tuple[str, int]]: """Returns the payload format and data view URI and ID from examples.""" assert examples.type is standard_artifacts.Examples, ( 'examples must be of type standard_artifacts.Examples') payload_format = examples_utils.get_payload_format(examples) if payload_format == example_gen_pb2.PayloadFormat.FORMAT_PROTO: data_view_uri = examples.get_string_custom_property( constants.DATA_VIEW_URI_PROPERTY_KEY) if data_view_uri: data_view_create_time = examples.get_int_custom_property( constants.DATA_VIEW_CREATE_TIME_KEY) return data_view_uri, data_view_create_time return None