Added overlapping_axes to api.md and __all__ in axis.py

stanford-crfm · Sep 6, 2024 · 00585c6 · 00585c6
1 parent 41a4fcf
commit 00585c6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/docs/api.md b/docs/api.md
@@ -34,6 +34,7 @@ Occasionally, an axis size can be inferred in some circumstances but not others.
 ::: haliax.axis.eliminate_axes
 ::: haliax.axis.without_axes
 ::: haliax.axis.selects_axis
+::: haliax.axis.overlapping_axes
 ::: haliax.axis.is_axis_compatible
 
 

diff --git a/src/haliax/axis.py b/src/haliax/axis.py
@@ -561,8 +561,9 @@ def replace_missing_with_ellipsis(ax1: AxisSelection, ax2: AxisSelection) -> Par
     "dslice",
     "dblock",
     "eliminate_axes",
-    "is_axis_compatible",
     "intersect_axes",
+    "is_axis_compatible",
+    "overlapping_axes",
     "replace_axis",
     "selects_axis",
     "union_axes",