Join the SIG TFX-Addons community and help make TFX even better!

tfx.types.Artifact

TFX artifact used for orchestration.

tfx.types.Artifact(
    mlmd_artifact_type: Optional[metadata_store_pb2.ArtifactType] = None
)

Used in the notebooks

Used in the tutorials

This is used for type-checking and inter-component communication. Currently, it wraps a tuple of (ml_metadata.proto.Artifact, ml_metadata.proto.ArtifactType) with additional property accessors for internal state.

A user may create a subclass of Artifact and override the TYPE_NAME property with the type for this artifact subclass. Users of the subclass may then omit the "type_name" field when construction the object.

A user may specify artifact type-specific properties for an Artifact subclass by overriding the PROPERTIES dictionary, as detailed below.

mlmd_artifact_type Proto message defining the underlying ArtifactType. Optional and intended for internal use.

artifact_type Type of the underlying mlmd artifact.
id Id of the underlying mlmd artifact.
mlmd_artifact Underlying mlmd artifact.
pipeline_name Name of the pipeline that produce the artifact.
state State of the underlying mlmd artifact.
type Type of the artifact.
type_id Type id of the underlying mlmd artifact.
type_name Type name of the underlying mlmd artifact.
uri Artifact URI.

Methods

get_float_custom_property

View source

get_float_custom_property(
    key: Text
) -> float

Gets a custom property of float type.

get_int_custom_property

View source

get_int_custom_property(
    key: Text
) -> int

Get a custom property of int type.

get_string_custom_property

View source

get_string_custom_property(
    key: Text
) -> Text

Get a custom property of string type.

has_custom_property

View source

has_custom_property(
    key: Text
) -> bool

set_float_custom_property

View source

set_float_custom_property(
    key: Text,
    value: float
)

Sets a custom property of float type.

set_int_custom_property

View source

set_int_custom_property(
    key: Text,
    value: int
)

Set a custom property of int type.

set_string_custom_property

View source

set_string_custom_property(
    key: Text,
    value: Text
)

Set a custom property of string type.

PROPERTIES None
TYPE_NAME None