From 21db0767fa50cdcff05e0ad4e1c17847fa18fa12 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Felix=20K=C3=B6hler?=
 <27728103+Ceyron@users.noreply.github.com>
Date: Wed, 10 Apr 2024 07:00:54 +0200
Subject: [PATCH] Fix shape in typing information

---
 exponax/_base_stepper.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/exponax/_base_stepper.py b/exponax/_base_stepper.py
index 5776597..54683bf 100644
--- 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