make_sequence_value_info alias for backwards compatibility (onnx#3603) (

onnx#3612) * Create alias make_sequence_value_info for backwards compatibility Signed-off-by: Rajeev Rao <[email protected]> * Add extra newline between functions to fix flake8 issue Signed-off-by: Rajeev Rao <[email protected]> * Add deprecation warnings for the make_sequence_value_info alias Signed-off-by: Rajeev Rao <[email protected]> * Update test release version to 1.9.101 Signed-off-by: Rajeev Rao <[email protected]> * Fix more flake8 whitespace issues Signed-off-by: Rajeev Rao <[email protected]>
jslhcl · Jul 28, 2021 · 04bb946 · 04bb946
1 parent a57bc99
commit 04bb946
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/onnx/helper.py b/onnx/helper.py
@@ -18,6 +18,7 @@
 from onnx.mapping import STORAGE_TENSOR_TYPE_TO_FIELD
 from typing import Text, Sequence, Any, Optional, Dict, Union, TypeVar, Callable, Tuple, List, cast
 import numpy as np  # type: ignore
+import warnings
 
 VersionRowType = Union[Tuple[Text, int, int, int], Tuple[Text, int, int, int, int]]
 VersionTableType = List[VersionRowType]
@@ -905,3 +906,16 @@ def make_training_info(algorithm, algorithm_bindings, initialization, initializa
             binding.value = v
 
     return training_info
+
+
+# For backwards compatibility
+def make_sequence_value_info(
+        name,  # type: Text
+        elem_type,  # type: int
+        shape,  # type: Optional[Sequence[Union[Text, int, None]]]
+        doc_string="",  # type: Text
+        elem_shape_denotation=None,  # type: Optional[List[Text]]
+):  # type: (...) -> ValueInfoProto
+    """Makes a Sequence[Tensors] ValueInfoProto based on the data type and shape."""
+    warnings.warn(str("`onnx.helper.make_sequence_value_info` is a deprecated alias for `onnx.helper.make_tensor_sequence_value_info`. To silence this warning, please use `make_tensor_sequence_value_info` for `TensorProto` sequences. Deprecated in ONNX v1.10.0, `onnx.helper.make_sequence_value_info alias` will be removed in an upcoming release."), DeprecationWarning, stacklevel=2)
+    return make_tensor_sequence_value_info(name, elem_type, shape, doc_string, elem_shape_denotation)