Adapt to new pytorch onnx exporter API for dictionary (#3160)

* Adapt to new torch export API for dictionary

Author: jiafatom

.circleci/config.yml

@@ -157,7 +157,7 @@ jobs:
             # need to install torchvision dependencies due to transitive imports
             pip install --user --progress-bar off --editable .
             pip install --user onnx
-            pip install --user -i https://test.pypi.org/simple/ ort-nightly==1.5.2.dev202012031
+            pip install --user onnxruntime
             python test/test_onnx.py
 
   binary_linux_wheel:

.circleci/config.yml.in

@@ -157,7 +157,7 @@ jobs:
             # need to install torchvision dependencies due to transitive imports
             pip install --user --progress-bar off --editable .
             pip install --user onnx
-            pip install --user -i https://test.pypi.org/simple/ ort-nightly==1.5.2.dev202012031
+            pip install --user onnxruntime
             python test/test_onnx.py
 
   binary_linux_wheel:

test/test_onnx.py

@@ -33,8 +33,12 @@ def run_model(self, model, inputs_list, tolerate_small_mismatch=False, do_consta
         model.eval()
 
         onnx_io = io.BytesIO()
+        if isinstance(inputs_list[0][-1], dict):
+            torch_onnx_input = inputs_list[0] + ({},)
+        else:
+            torch_onnx_input = inputs_list[0]
         # export to onnx with the first input
-        torch.onnx.export(model, inputs_list[0], onnx_io,
+        torch.onnx.export(model, torch_onnx_input, onnx_io,
                           do_constant_folding=do_constant_folding, opset_version=_onnx_opset_version,
                           dynamic_axes=dynamic_axes, input_names=input_names, output_names=output_names)
         # validate the exported model with onnx runtime