[r0.30.0 cherry-pick] commit_id=[2736197] (#4011)

tensorflow · Jul 9, 2021 · b364cd1 · b364cd1
1 parent 8a5c348
commit b364cd1
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/tfx/orchestration/kubeflow/v2/step_builder.py b/tfx/orchestration/kubeflow/v2/step_builder.py
@@ -586,8 +586,9 @@ def _build_resolver_for_latest_blessed_model(
     artifact_queries = {}
     query_filter = (
         'schema_title="{type}" AND '
-        'state={state} AND name={{$.inputs.artifacts["{input_key}"]'
-        '.metadata.{property_key}.number_value}}').format(
+        'state={state} AND '
+        'name="{{{{$.inputs.artifacts[\'{input_key}\']'
+        '.metadata[\'{property_key}\']}}}}"').format(
             type=compiler_utils.get_artifact_title(standard_artifacts.Model),
             state=metadata_store_pb2.Artifact.State.Name(
                 metadata_store_pb2.Artifact.LIVE),

diff --git a/tfx/orchestration/kubeflow/v2/testdata/expected_full_taxi_pipeline_job.json b/tfx/orchestration/kubeflow/v2/testdata/expected_full_taxi_pipeline_job.json
@@ -57,7 +57,7 @@
           "resolver": {
             "outputArtifactQueries": {
               "model": {
-                "filter": "schema_title=\"tfx.Model\" AND state=LIVE AND name={$.inputs.artifacts[\"input\"].metadata.current_model_id.number_value}"
+                "filter": "schema_title=\"tfx.Model\" AND state=LIVE AND name=\"{{$.inputs.artifacts['input'].metadata['current_model_id']}}\""
               }
             }
           }

diff --git a/tfx/orchestration/kubeflow/v2/testdata/expected_latest_blessed_model_resolver_executor.pbtxt b/tfx/orchestration/kubeflow/v2/testdata/expected_latest_blessed_model_resolver_executor.pbtxt
@@ -21,7 +21,7 @@ executors {
       output_artifact_queries {
         key: "model"
         value {
-          filter: "schema_title=\"tfx.Model\" AND state=LIVE AND name={$.inputs.artifacts[\"input\"].metadata.current_model_id.number_value}"
+          filter: "schema_title=\"tfx.Model\" AND state=LIVE AND name=\"{{$.inputs.artifacts['input'].metadata['current_model_id']}}\""
         }
       }
     }