Merge branch 'microsoft:main' into doc-script-const-1

microsoft · Jan 9, 2025 · e906138 · e906138
2 parents 4804d5d + 646116c
commit e906138
Show file tree

Hide file tree

Showing 84 changed files with 405 additions and 3,545 deletions.
diff --git a/noxfile.py b/noxfile.py
@@ -27,7 +27,7 @@
     "pytest!=7.1.0",
     "pyyaml",
     "types-PyYAML",
-    "typing_extensions",
+    "typing_extensions>=4.10",
     "ml-dtypes",
 )
 ONNX = "onnx==1.17"

diff --git a/onnxscript/converter.py b/onnxscript/converter.py
@@ -800,6 +800,9 @@ def translate_slice(slice_expr: ast.Slice) -> tuple[str, str, str]:
         non_scalar_indices.extend(scalar_indices)
         if non_scalar_indices:
             last_axis, _ = non_scalar_indices[-1]
+        else:
+            # TODO(justinchuby): Clarify what last_axis should be when non_scalar_indices is False
+            last_axis = None
         for axis, index_expr in non_scalar_indices:
             index_value = self._translate_expr(index_expr)
             axis_attr = self._make_onnx_attr("axis", axis)

diff --git a/onnxscript/diagnostics/infra/__init__.py b/onnxscript/diagnostics/infra/__init__.py
diff --git a/onnxscript/diagnostics/infra/_infra.py b/onnxscript/diagnostics/infra/_infra.py