import onnx
from onnx import TensorProto
from onnx.helper import (
    make_model, make_node, make_graph, make_attribute,
    make_tensor_value_info)
from onnx.checker import check_model
import onnxruntime

print(onnx.__version__, onnxruntime.__version__)

def create_SqUsq_model(target_shape) -> bytes:
    X = make_tensor_value_info("X", TensorProto.FLOAT, target_shape)
    Y = make_tensor_value_info("Y", TensorProto.FLOAT, target_shape)
    node1 = make_node("Squeeze", ["X"], ["A"])
    node2 = make_node("Unsqueeze", ["A"], ["Y"])
    node2.attribute.append(make_attribute("axes", [1]))
    graph = make_graph([node1, node2], "sample", [X], [Y])
    model = make_model(graph, opset_imports=[onnx.helper.make_operatorsetid("", 12)])
    check_model(model)
    return model.SerializeToString()

model = create_SqUsq_model([5, 1])
sess = onnxruntime.InferenceSession(model, providers=["CPUExecutionProvider"]) # it works

model = create_SqUsq_model([5, None])
sess = onnxruntime.InferenceSession(model, providers=["CPUExecutionProvider"]) # raise error

# outputs:
# 1.16.2 1.19.0
# Traceback (most recent call last):
#   File "C:\Users\yoshi\Desktop\vvinfer27\gen.py", line 23, in <module>
#     sess = onnxruntime.InferenceSession(model, providers=["CPUExecutionProvider"])
#            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
#   File "C:\Users\yoshi\miniconda3\envs\onnx\Lib\site-packages\onnxruntime\capi\onnxruntime_inference_collection.py", line 419, in __init__
#     self._create_inference_session(providers, provider_options, disabled_optimizers)
#   File "C:\Users\yoshi\miniconda3\envs\onnx\Lib\site-packages\onnxruntime\capi\onnxruntime_inference_collection.py", line 482, in _create_inference_session
#     sess = C.InferenceSession(session_options, self._model_bytes, False, self._read_config_from_model)
#            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
# onnxruntime.capi.onnxruntime_pybind11_state.Fail: [ONNXRuntimeError] : 1 : FAIL : Node () Op (Unsqueeze) [ShapeInferenceError] values in 'axes' are beyond the bounds of the computed output shape