lint

Author: xadupre

_unittests/ut_tasks/test_tasks.py

@@ -65,14 +65,8 @@ def test_automatic_speech_recognition_float32(self):
                 "cache_position": {0: "seq_length"},
                 "encoder_outputs": [{0: "batch"}],
                 "past_key_values": [
-                    [
-                        [{0: "batch"}, {0: "batch"}],
-                        [{0: "batch"}, {0: "batch"}],
-                    ],
-                    [
-                        [{0: "batch"}, {0: "batch"}],
-                        [{0: "batch"}, {0: "batch"}],
-                    ],
+                    [[{0: "batch"}, {0: "batch"}], [{0: "batch"}, {0: "batch"}]],
+                    [[{0: "batch"}, {0: "batch"}], [{0: "batch"}, {0: "batch"}]],
                 ],
             },
             ds,
@@ -126,14 +120,8 @@ def test_automatic_speech_recognition_float16(self):
                 "cache_position": {0: "seq_length"},
                 "encoder_outputs": [{0: "batch"}],
                 "past_key_values": [
-                    [
-                        [{0: "batch"}, {0: "batch"}],
-                        [{0: "batch"}, {0: "batch"}],
-                    ],
-                    [
-                        [{0: "batch"}, {0: "batch"}],
-                        [{0: "batch"}, {0: "batch"}],
-                    ],
+                    [[{0: "batch"}, {0: "batch"}], [{0: "batch"}, {0: "batch"}]],
+                    [[{0: "batch"}, {0: "batch"}], [{0: "batch"}, {0: "batch"}]],
                 ],
             },
             ds,

requirements-dev.txt

@@ -8,7 +8,7 @@ matplotlib
 onnx-array-api>=0.3.1
 onnx
 git+https://github.com/onnx/ir-py.git
-git+https://github.com/microsoft/onnxscript.git
+onnxscript
 openpyxl
 packaging
 pandas