address comments

googleapis · Nov 15, 2021 · df9c883 · df9c883
1 parent a88a618
commit df9c883
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: