class _MyArtifact4(artifact.Artifact): TYPE_NAME = 'MyTypeName4' TYPE_ANNOTATION = Dataset PROPERTIES = { 'span': artifact.Property(type=artifact.PropertyType.INT), 'split_names': artifact.Property(type=artifact.PropertyType.STRING), }
class _MyArtifact(artifact.Artifact): TYPE_NAME = 'MyTypeName' PROPERTIES = { 'int1': artifact.Property(type=artifact.PropertyType.INT), 'int2': artifact.Property(type=artifact.PropertyType.INT), 'string1': artifact.Property(type=artifact.PropertyType.STRING), 'string2': artifact.Property(type=artifact.PropertyType.STRING), }
class _MyBadArtifact(artifact.Artifact): TYPE_NAME = 'TestBadType' PROPERTIES = { 'int1': artifact.Property(type=artifact.PropertyType.INT), }
class _MyArtifact(artifact.Artifact): TYPE_NAME = 'MyTypeName' PROPERTIES = { 'int1': artifact.Property(type=artifact.PropertyType.INT), 'int2': artifact.Property(type=artifact.PropertyType.INT), 'float1': artifact.Property(type=artifact.PropertyType.FLOAT), 'float2': artifact.Property(type=artifact.PropertyType.FLOAT), 'string1': artifact.Property(type=artifact.PropertyType.STRING), 'string2': artifact.Property(type=artifact.PropertyType.STRING), } _MyArtifact2 = artifact._ArtifactType( # pylint: disable=invalid-name name='MyTypeName2', properties={ 'int1': artifact.Property(type=artifact.PropertyType.INT), 'int2': artifact.Property(type=artifact.PropertyType.INT), 'float1': artifact.Property(type=artifact.PropertyType.FLOAT), 'float2': artifact.Property(type=artifact.PropertyType.FLOAT), 'string1': artifact.Property(type=artifact.PropertyType.STRING), 'string2': artifact.Property(type=artifact.PropertyType.STRING), }) _mlmd_artifact_type = metadata_store_pb2.ArtifactType() json_format.Parse( json.dumps({ 'name': 'MyTypeName3', 'properties': { 'int1': 'INT', 'int2': 'INT', 'float1': 'DOUBLE',
class _MyArtifact(artifact.Artifact): TYPE_NAME = 'ArtifactUtilsTypeName' PROPERTIES = { 'dummy_int': artifact.Property(artifact.PropertyType.INT), 'dummy_string': artifact.Property(artifact.PropertyType.STRING), }
class _MyType(artifact.Artifact): TYPE_NAME = "MyTypeName" PROPERTIES = { "string_value": artifact.Property(artifact.PropertyType.STRING), }
from tfx.types import artifact, standard_artifacts PARTITIONS_PROPERTY = artifact.Property(type=artifact.PropertyType.STRING) class Datasets(artifact.Artifact): TYPE_NAME = 'Datasets' class PartitionedExamples(artifact.Artifact): TYPE_NAME = 'Partitions' PROPERTIES = { 'span': standard_artifacts.SPAN_PROPERTY, 'split_names': standard_artifacts.SPLIT_NAMES_PROPERTY, 'partitions': PARTITIONS_PROPERTY }
class _MyArtifact(artifact.Artifact): TYPE_NAME = 'MyTypeName' PROPERTIES = { 'int1': artifact.Property(type=artifact.PropertyType.INT), }
class _MyType(artifact.Artifact): TYPE_NAME = 'MyTypeName' PROPERTIES = { 'string_value': artifact.Property(artifact.PropertyType.STRING), }
class _ArtifactWithInvalidAnnotation(artifact.Artifact): TYPE_NAME = 'InvalidAnnotationArtifact' TYPE_ANNOTATION = artifact.Artifact PROPERTIES = { 'int1': artifact.Property(type=artifact.PropertyType.INT), }
class _MyArtifact(artifact.Artifact): TYPE_NAME = 'MyTypeName' PROPERTIES = { 'int1': artifact.Property(type=artifact.PropertyType.INT), 'int2': artifact.Property(type=artifact.PropertyType.INT), 'float1': artifact.Property(type=artifact.PropertyType.FLOAT), 'float2': artifact.Property(type=artifact.PropertyType.FLOAT), 'string1': artifact.Property(type=artifact.PropertyType.STRING), 'string2': artifact.Property(type=artifact.PropertyType.STRING), } _MyArtifact2 = artifact._ArtifactType( # pylint: disable=invalid-name name='MyTypeName2', properties={ 'int1': artifact.Property(type=artifact.PropertyType.INT), 'int2': artifact.Property(type=artifact.PropertyType.INT), 'float1': artifact.Property(type=artifact.PropertyType.FLOAT), 'float2': artifact.Property(type=artifact.PropertyType.FLOAT), 'string1': artifact.Property(type=artifact.PropertyType.STRING), 'string2': artifact.Property(type=artifact.PropertyType.STRING), 'jsonvalue_string': artifact.Property(type=artifact.PropertyType.JSON_VALUE), 'jsonvalue_dict': artifact.Property(type=artifact.PropertyType.JSON_VALUE), 'jsonvalue_int': artifact.Property(type=artifact.PropertyType.JSON_VALUE), 'jsonvalue_float': artifact.Property(type=artifact.PropertyType.JSON_VALUE), 'jsonvalue_list': artifact.Property(type=artifact.PropertyType.JSON_VALUE),