From df9c883734169920f15ec80efcd966846bfcef89 Mon Sep 17 00:00:00 2001 From: Yaqi Date: Mon, 15 Nov 2021 11:29:43 -0800 Subject: [PATCH] address comments --- google/cloud/aiplatform/utils/pipeline_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/google/cloud/aiplatform/utils/pipeline_utils.py b/google/cloud/aiplatform/utils/pipeline_utils.py index 506dbdff26..5a09a98f18 100644 --- a/google/cloud/aiplatform/utils/pipeline_utils.py +++ b/google/cloud/aiplatform/utils/pipeline_utils.py @@ -138,7 +138,7 @@ def build(self) -> Dict[str, Any]: def _get_vertex_value( self, name: str, value: Union[int, float, str, bool, list, dict] - ) -> Union[Dict[str, Any], int, float, str, bool, list, dict]: + ) -> Union[int, float, str, bool, list, dict]: """Converts primitive values into Vertex pipeline Value proto message. Args: