merge, undo setup.py change

kubeflow · Sep 26, 2020 · 1b58d6b · 1b58d6b
1 parent 573c9f3
commit 1b58d6b
Show file tree

Hide file tree

Showing 6 changed files with 4 additions and 477 deletions.
diff --git a/sdk/python/kfp/dsl/_component_bridge.py b/sdk/python/kfp/dsl/_component_bridge.py
@@ -19,7 +19,7 @@
 from ..components._naming import _sanitize_python_function_name, generate_unique_name_conversion_table
 from .. import dsl
 from kfp.dsl import ir_types
-from kfp.ir import pipeline_spec_pb2
+from kfp.v2.proto import pipeline_spec_pb2
 
 def _create_container_op_from_component_and_arguments(
     component_spec: ComponentSpec,

diff --git a/sdk/python/kfp/dsl/ir_types.py b/sdk/python/kfp/dsl/ir_types.py
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from kfp.ir import pipeline_spec_pb2
+from kfp.v2.proto import pipeline_spec_pb2
 
  # TODO: support more artifact types
 _artifact_types_mapping = {