fix2

kubeflow · Dec 8, 2021 · 18d796b · 18d796b
1 parent a2aa571
commit 18d796b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 9 deletions.
diff --git a/samples/test/lightweight_python_functions_v2_pipeline_test.py b/samples/test/lightweight_python_functions_v2_pipeline_test.py
@@ -39,7 +39,6 @@ def verify(run: kfp_server_api.ApiRun, mlmd_connection_config, **kwargs):
     t.assertEqual(
         {
             'inputs': {
-                'artifacts': [],
                 'parameters': {
                     'message': 'message',
                 }
@@ -93,8 +92,8 @@ def verify(run: kfp_server_api.ApiRun, mlmd_connection_config, **kwargs):
                 'parameters': {
                     'input_bool': True,
                     'input_dict': {
-                        "A": 1,
-                        "B": 2
+                        "A": 1.0,
+                        "B": 2.0,
                     },
                     'input_list': ["a", "b", "c"],
                     'message': 'message'
@@ -110,7 +109,6 @@ def verify(run: kfp_server_api.ApiRun, mlmd_connection_config, **kwargs):
                     'name': 'model',
                     'type': 'system.Model'
                 }],
-                'parameters': {}
             },
             'type': 'system.ContainerExecution',
             'state': Execution.State.COMPLETE,

diff --git a/samples/test/metrics_visualization_v2_test.py b/samples/test/metrics_visualization_v2_test.py
@@ -153,10 +153,6 @@ def verify(t: unittest.TestCase, run: kfp_server_api.ApiRun,
 
     t.assertEqual(
         {
-            'inputs': {
-                'artifacts': [],
-                'parameters': {}
-            },
             'name': 'digit-classification',
             'outputs': {
                 'artifacts': [{
@@ -167,7 +163,6 @@ def verify(t: unittest.TestCase, run: kfp_server_api.ApiRun,
                     'name': 'metrics',
                     'type': 'system.Metrics'
                 }],
-                'parameters': {}
             },
             'type': 'system.ContainerExecution',
             'state': Execution.State.COMPLETE,