示例#1
0
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),
    }
示例#2
0
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),
    }
示例#3
0
class _MyBadArtifact(artifact.Artifact):
    TYPE_NAME = 'TestBadType'
    PROPERTIES = {
        'int1': artifact.Property(type=artifact.PropertyType.INT),
    }
示例#4
0
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',
示例#5
0
class _MyArtifact(artifact.Artifact):
    TYPE_NAME = 'ArtifactUtilsTypeName'
    PROPERTIES = {
        'dummy_int': artifact.Property(artifact.PropertyType.INT),
        'dummy_string': artifact.Property(artifact.PropertyType.STRING),
    }
示例#6
0
class _MyType(artifact.Artifact):
  TYPE_NAME = "MyTypeName"
  PROPERTIES = {
      "string_value": artifact.Property(artifact.PropertyType.STRING),
  }
示例#7
0
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),
  }
示例#9
0
class _MyType(artifact.Artifact):
  TYPE_NAME = 'MyTypeName'
  PROPERTIES = {
      'string_value': artifact.Property(artifact.PropertyType.STRING),
  }
示例#10
0
class _ArtifactWithInvalidAnnotation(artifact.Artifact):
    TYPE_NAME = 'InvalidAnnotationArtifact'
    TYPE_ANNOTATION = artifact.Artifact
    PROPERTIES = {
        'int1': artifact.Property(type=artifact.PropertyType.INT),
    }
示例#11
0
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),