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