Fix v1 tests

kubeflow · Aug 19, 2021 · 13dd8e7 · 13dd8e7
1 parent 5b7233d
commit 13dd8e7
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 14 deletions.
diff --git a/sdk/python/kfp/compiler/testdata/v2_compatible_two_step_pipeline.yaml b/sdk/python/kfp/compiler/testdata/v2_compatible_two_step_pipeline.yaml
@@ -101,7 +101,7 @@ spec:
           "INT"}, "uri": {"type": "STRING"}}, "inputArtifacts": {}, "outputParameters":
           {"output_parameter_one": {"type": "INT", "path": "/tmp/outputs/output_parameter_one/data"}},
           "outputArtifacts": {"output_dataset_one": {"schemaTitle": "system.Dataset",
-          "instanceSchema": "", "metadataPath": "/tmp/outputs/output_dataset_one/data"}}}'}
+          "instanceSchema": "", "schemaVersion": "0.0.1", "metadataPath": "/tmp/outputs/output_dataset_one/data"}}}'}
       envFrom:
       - configMapRef: {name: metadata-grpc-configmap, optional: true}
       image: python:3.7
@@ -200,9 +200,9 @@ spec:
       - {name: KFP_V2_IMAGE, value: 'python:3.7'}
       - {name: KFP_V2_RUNTIME_INFO, value: '{"inputParameters": {"num_steps": {"type":
           "INT"}}, "inputArtifacts": {"dataset": {"metadataPath": "/tmp/inputs/dataset/data",
-          "schemaTitle": "system.Dataset", "instanceSchema": ""}}, "outputParameters":
+          "schemaTitle": "system.Dataset", "instanceSchema": "", "schemaVersion": "0.0.1"}}, "outputParameters":
           {}, "outputArtifacts": {"model": {"schemaTitle": "system.Model", "instanceSchema":
-          "", "metadataPath": "/tmp/outputs/model/data"}}}'}
+          "", "schemaVersion": "0.0.1", "metadataPath": "/tmp/outputs/model/data"}}}'}
       envFrom:
       - configMapRef: {name: metadata-grpc-configmap, optional: true}
       image: python:3.7

diff --git a/sdk/python/kfp/compiler/testdata/v2_compatible_two_step_pipeline_with_custom_launcher.yaml b/sdk/python/kfp/compiler/testdata/v2_compatible_two_step_pipeline_with_custom_launcher.yaml
@@ -101,7 +101,7 @@ spec:
           "INT"}, "uri": {"type": "STRING"}}, "inputArtifacts": {}, "outputParameters":
           {"output_parameter_one": {"type": "INT", "path": "/tmp/outputs/output_parameter_one/data"}},
           "outputArtifacts": {"output_dataset_one": {"schemaTitle": "system.Dataset",
-          "instanceSchema": "", "metadataPath": "/tmp/outputs/output_dataset_one/data"}}}'}
+          "instanceSchema": "", "schemaVersion": "0.0.1", "metadataPath": "/tmp/outputs/output_dataset_one/data"}}}'}
       envFrom:
       - configMapRef: {name: metadata-grpc-configmap, optional: true}
       image: python:3.7
@@ -200,9 +200,9 @@ spec:
       - {name: KFP_V2_IMAGE, value: 'python:3.7'}
       - {name: KFP_V2_RUNTIME_INFO, value: '{"inputParameters": {"num_steps": {"type":
           "INT"}}, "inputArtifacts": {"dataset": {"metadataPath": "/tmp/inputs/dataset/data",
-          "schemaTitle": "system.Dataset", "instanceSchema": ""}}, "outputParameters":
+          "schemaTitle": "system.Dataset", "instanceSchema": "", "schemaVersion": "0.0.1"}}, "outputParameters":
           {}, "outputArtifacts": {"model": {"schemaTitle": "system.Model", "instanceSchema":
-          "", "metadataPath": "/tmp/outputs/model/data"}}}'}
+          "", "schemaVersion": "0.0.1", "metadataPath": "/tmp/outputs/model/data"}}}'}
       envFrom:
       - configMapRef: {name: metadata-grpc-configmap, optional: true}
       image: python:3.7

diff --git a/sdk/python/tests/compiler/compiler_tests.py b/sdk/python/tests/compiler/compiler_tests.py
@@ -487,33 +487,33 @@ def my_pipeline(memory: str, cpu: str):
       template = name_to_template[main_dag_tasks[0]['template']]
 
       self.assertEqual(template['podSpecPatch'], '{"containers": [{"name": "main", "resources": {"requests": {"cpu": "{{inputs.parameters.memory}}"}}}]}')
-      
+
   def test_py_runtime_gpu_request(self):
       """Test GPU request."""
 
       def my_pipeline(nbr_gpus: int, gpu_vendor: str):
         some_op().set_gpu_limit(nbr_gpus, gpu_vendor)
-      
+
       workflow = kfp.compiler.Compiler()._create_workflow(my_pipeline)
       name_to_template = {template['name']: template for template in workflow['spec']['templates']}
       main_dag_tasks = name_to_template[workflow['spec']['entrypoint']]['dag']['tasks']
       template = name_to_template[main_dag_tasks[0]['template']]
 
       self.assertEqual(template['podSpecPatch'], '{"containers": [{"name": "main", "resources": {"limits": {"{{inputs.parameters.gpu_vendor}}": "{{inputs.parameters.nbr_gpus}}"}}}]}')
-      
+
   def test_py_runtime_node_selection(self):
       """Test node selection request."""
 
       def my_pipeline(constrain_type: str, constrain_value: str):
         some_op().add_node_selector_constraint(constrain_type, constrain_value)
-      
+
       workflow = kfp.compiler.Compiler()._create_workflow(my_pipeline)
       name_to_template = {template['name']: template for template in workflow['spec']['templates']}
       main_dag_tasks = name_to_template[workflow['spec']['entrypoint']]['dag']['tasks']
       template = name_to_template[main_dag_tasks[0]['template']]
 
       self.assertEqual(template['podSpecPatch'], '{"nodeSelector": [{"{{inputs.parameters.constrain_type}}": "{{inputs.parameters.constrain_value}}"}]}')
-     
+
 
   def test_py_retry_policy_invalid(self):
       def my_pipeline():

diff --git a/sdk/python/tests/compiler/testdata/uri_artifacts.yaml b/sdk/python/tests/compiler/testdata/uri_artifacts.yaml
@@ -154,7 +154,7 @@ spec:
       - {name: KFP_V2_IMAGE, value: 'google/cloud-sdk:slim'}
       - {name: KFP_V2_RUNTIME_INFO, value: '{"inputParameters": {}, "inputArtifacts":
           {"input_gcs_path": {"metadataPath": "/tmp/inputs/input_gcs_path/data", "schemaTitle":
-          "system.Artifact", "instanceSchema": ""}}, "outputParameters": {}, "outputArtifacts":
+          "system.Artifact", "instanceSchema": "", "schemaVersion": "0.0.1"}}, "outputParameters": {}, "outputArtifacts":
           {}}'}
       envFrom:
       - configMapRef: {name: metadata-grpc-configmap, optional: true}
@@ -221,7 +221,7 @@ spec:
       - {name: KFP_V2_IMAGE, value: 'google/cloud-sdk:slim'}
       - {name: KFP_V2_RUNTIME_INFO, value: '{"inputParameters": {}, "inputArtifacts":
           {"input_gcs_path": {"metadataPath": "/tmp/inputs/input_gcs_path/data", "schemaTitle":
-          "system.Artifact", "instanceSchema": ""}}, "outputParameters": {}, "outputArtifacts":
+          "system.Artifact", "instanceSchema": "", "schemaVersion": "0.0.1"}}, "outputParameters": {}, "outputArtifacts":
           {}}'}
       envFrom:
       - configMapRef: {name: metadata-grpc-configmap, optional: true}
@@ -288,7 +288,7 @@ spec:
       - {name: KFP_V2_IMAGE, value: 'google/cloud-sdk:slim'}
       - {name: KFP_V2_RUNTIME_INFO, value: '{"inputParameters": {}, "inputArtifacts":
           {"input_gcs_path": {"metadataPath": "/tmp/inputs/input_gcs_path/data", "schemaTitle":
-          "system.Artifact", "instanceSchema": ""}}, "outputParameters": {}, "outputArtifacts":
+          "system.Artifact", "instanceSchema": "", "schemaVersion": "0.0.1"}}, "outputParameters": {}, "outputArtifacts":
           {}}'}
       envFrom:
       - configMapRef: {name: metadata-grpc-configmap, optional: true}