Python Artifact.get_int_custom_property Examples

Example #1

File: kubeflow_v2_entrypoint_utils.py Project: suryaavala/tfx

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))

Example #2

File: tfxio_utils.py Project: htahir1/tfx

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