address comments

Author: ji-yaqi

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: