Python PipelineRuntimeSpec Examples

Example #1

File: outputs_utils_test.py Project: jay90099/tfx

 def setUp(self):
     super().setUp()
     pipeline_runtime_spec = pipeline_pb2.PipelineRuntimeSpec()
     pipeline_runtime_spec.pipeline_root.field_value.string_value = self.tmp_dir
     pipeline_runtime_spec.pipeline_run_id.field_value.string_value = (
         'test_run_0')
     self._pipeline_runtime_spec = pipeline_runtime_spec

Example #2

File: outputs_utils_test.py Project: kp425/tfx

    def setUp(self):
        super().setUp()
        pipeline_runtime_spec = pipeline_pb2.PipelineRuntimeSpec()
        pipeline_runtime_spec.pipeline_root.field_value.string_value = self.tmp_dir
        pipeline_runtime_spec.pipeline_run_id.field_value.string_value = (
            'test_run_0')

        self._output_resolver = outputs_utils.OutputsResolver(
            pipeline_node=_PIPLINE_NODE,
            pipeline_info=_PIPELING_INFO,
            pipeline_runtime_spec=pipeline_runtime_spec)

Example #3

  id: "my_pipeline"
""", pipeline_pb2.PipelineInfo())

_PIPELINE_RUNTIME_SPEC = text_format.Parse(
    """
  pipeline_root {
    field_value {
      string_value: "/tmp"
    }
  }
  pipeline_run_id {
    field_value {
      string_value: "my_run_id"
    }
  }
""", pipeline_pb2.PipelineRuntimeSpec())

_PIPLINE_NODE = text_format.Parse(
    """
  node_info {
    id: "test_node"
  }
  inputs {
    inputs {
      key: "model"
      value {
        channels {
          artifact_query {
            type {
              name: "Model"
            }