Fix shape in typing information

Ceyron · Apr 10, 2024 · 21db076 · 21db076
1 parent d890bce
commit 21db076
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/exponax/_base_stepper.py b/exponax/_base_stepper.py
@@ -159,7 +159,7 @@ def __init__(
     def _build_linear_operator(
         self,
         derivative_operator: Complex[Array, "D ... (N//2)+1"],
-    ) -> Complex[Array, "D ... (N//2)+1"]:
+    ) -> Complex[Array, "C ... (N//2)+1"]:
         """
         Assemble the L operator in Fourier space.
 
@@ -169,7 +169,7 @@ def _build_linear_operator(
               N//2+1).
 
         **Returns:**
-            - `L`: The linear operator, shape `( D, ..., N//2+1 )`.
+            - `L`: The linear operator, shape `( C, ..., N//2+1 )`.
         """
         pass