Add missing matmul to accessor

Author: krokosik

src/xarray_einstats/accessors.py

@@ -12,6 +12,7 @@
     eigvals,
     eigvalsh,
     inv,
+    matmul,
     matrix_power,
     matrix_rank,
     matrix_transpose,
@@ -40,6 +41,10 @@ def matrix_power(self, n, dims=None, **kwargs):
         """Call :func:`xarray_einstats.linalg.matrix_power` on this DataArray."""
         return matrix_power(self._obj, n, dims=dims, **kwargs)
 
+    def matmul(self, other, dims=None, **kwargs):
+        """Call :func:`xarray_einstats.linalg.matmul` with this DataArray as ``a/da``."""
+        return matmul(self._obj, other, dims=dims, **kwargs)
+
     def cholesky(self, dims=None, **kwargs):
         """Call :func:`xarray_einstats.linalg.cholesky` on this DataArray."""
         return cholesky(self._obj, dims=dims, **kwargs)

src/xarray_einstats/accessors.pyi

@@ -13,6 +13,7 @@ from .linalg import (
     eigvals,
     eigvalsh,
     inv,
+    matmul,
     matrix_power,
     matrix_rank,
     matrix_transpose,
@@ -28,6 +29,7 @@ class LinAlgAccessor:
     def __init__(self, xarray_obj: Incomplete) -> None: ...
     def matrix_transpose(self, dims: Incomplete) -> None: ...
     def matrix_power(self, n: Incomplete, dims: Incomplete = ..., **kwargs: Incomplete) -> None: ...
+    def matmul(self, other: Incomplete, dims: Incomplete = ..., **kwargs: Incomplete) -> None: ...
     def cholesky(self, dims: Incomplete = ..., **kwargs: Incomplete) -> None: ...
     def qr(
         self,