Fix type variable names

Author: gjbex

source-code/typing/numpy_typing.py

@@ -6,9 +6,12 @@
 import numpy.typing as npt
 
 
-def generate_input(a: float | np.float32 | np.float64,
-                   b: float | np.float32 | np.float64,
-                   n: int | np.int32 | np.int64) -> npt.NDArray[np.float64]:
+type Float = float | np.float32 | np.float64
+type Int = int | np.int32 | np.int64
+type Vector = npt.NDArray[np.float64]
+
+
+def generate_input(a: Float, b: Float, n: Int) -> Vector:
     '''Generate x-values in given range
 
     Parameters
@@ -28,12 +31,12 @@ def generate_input(a: float | np.float32 | np.float64,
     return np.linspace(a, b, n)
 
 
-def gaussian(x: npt.NDArray[np.float64], mu: float, sigma: float) -> npt.NDArray[np.float64]:
-    y: npt.NDArray[np.float64] = np.exp(-0.5*(x - mu)**2/sigma)/np.sqrt(2.0*np.pi*sigma)
+def gaussian(x: Vector, mu: Float, sigma: Float) -> Vector:
+    y: Vector = np.exp(-0.5*(x - mu)**2/sigma)/np.sqrt(2.0*np.pi*sigma)
     return y
 
 
-def plot_function(x: npt.NDArray[np.float64], y: npt.NDArray[np.float64]) -> None:
+def plot_function(x: Vector, y: Vector) -> None:
     plt.plot(x, y)
     plt.show()
     return