adding to linearmap operation

src/aggforce/map/jaxlinearmap.py

@@ -1,4 +1,5 @@
 """Extends LinearMaps for Jax operations."""
+
 from typing import overload, TypeVar
 from jax import Array
 import jax.numpy as jnp
@@ -26,12 +27,10 @@ def jax_standard_matrix(self) -> Array:
         return jnp.asarray(self.standard_matrix)
 
     @overload
-    def __call__(self, points: NDArray) -> NDArray:
-        ...
+    def __call__(self, points: NDArray) -> NDArray: ...
 
     @overload
-    def __call__(self, points: Array) -> Array:
-        ...
+    def __call__(self, points: Array) -> Array: ...
 
     def __call__(self, points: ArrT) -> ArrT:
         r"""Apply map to a particular form of 3-dim array.
@@ -65,12 +64,10 @@ def __call__(self, points: ArrT) -> ArrT:
             return transformed
 
     @overload
-    def flat_call(self, flattened: NDArray) -> NDArray:
-        ...
+    def flat_call(self, flattened: NDArray) -> NDArray: ...
 
     @overload
-    def flat_call(self, flattened: Array) -> Array:
-        ...
+    def flat_call(self, flattened: Array) -> Array: ...
 
     def flat_call(self, flattened: ArrT) -> ArrT:
         """Apply map to pre-flattened array.
@@ -129,3 +126,7 @@ def __add__(self, lm: "LinearMap", /) -> "JLinearMap":
     def from_linearmap(cls, lm: LinearMap, /) -> "JLinearMap":
         """Create JLinearMap from LinearMap."""
         return JLinearMap(mapping=lm.standard_matrix)
+
+    def to_linearmap(self) -> LinearMap:
+        """Create normal LinearMap from the current object."""
+        return LinearMap(mapping=self.standard_matrix)