fix: Update typing checks to include Python3.6 deprecated types. (#6520)

kubeflow · Sep 8, 2021 · b6cb0b4 · b6cb0b4
1 parent 9ed77f5
commit b6cb0b4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/components/google-cloud/google_cloud_pipeline_components/aiplatform/utils.py b/components/google-cloud/google_cloud_pipeline_components/aiplatform/utils.py
@@ -16,7 +16,7 @@
 import collections
 import inspect
 import json
-from typing import Any, Callable, Dict, List, Optional, Tuple, Union
+from typing import Any, Callable, Dict, List, Optional, Sequence, Tuple, Union
 import tempfile
 import docstring_parser
 
@@ -136,7 +136,7 @@ def is_serializable_to_json(annotation: Any) -> bool:
     Returns:
         True if serializable to json.
     """
-    serializable_types = (dict, list, collections.abc.Sequence)
+    serializable_types = (dict, list, collections.abc.Sequence, Dict, Sequence)
     return getattr(annotation, '__origin__', None) in serializable_types