From 2cf0693070f74790c9209a00182724dd6ed818d3 Mon Sep 17 00:00:00 2001
From: Arity-T <tish.mainmail@gmail.com>
Date: Wed, 15 Oct 2025 16:43:05 +0300
Subject: [PATCH] =?UTF-8?q?=D0=9E=D0=BF=D0=B5=D1=87=D0=B0=D1=82=D0=BA?=
 =?UTF-8?q?=D0=B0=20=D0=B2=20=D1=82=D0=B8=D0=BF=D0=B0=D1=85=20=D0=B2=20lab?=
 =?UTF-8?q?2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 lab2/gen.py  | 4 ++--
 lab2/main.py | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/lab2/gen.py b/lab2/gen.py
index f6fab19..1ccf957 100644
--- a/lab2/gen.py
+++ b/lab2/gen.py
@@ -16,7 +16,7 @@ from numpy.typing import NDArray
 type Chromosome = NDArray[np.float64]
 type Population = list[Chromosome]
 type Fitnesses = NDArray[np.float64]
-type FitnessFn = Callable[[Chromosome], Fitnesses]
+type FitnessFn = Callable[[Chromosome], np.float64]
 type CrossoverFn = Callable[[Chromosome, Chromosome], tuple[Chromosome, Chromosome]]
 type MutationFn = Callable[[Chromosome], Chromosome]
 
@@ -185,7 +185,7 @@ def plot_fitness_surface(
     x_max: Chromosome,
     ax: Axes3D,
     num_points: int = 100,
-) -> None:
+):
     """Рисует поверхность функции фитнеса в 3D."""
     assert (
         x_min.shape == x_max.shape == (2,)
diff --git a/lab2/main.py b/lab2/main.py
index 0f3432e..5099c1a 100644
--- a/lab2/main.py
+++ b/lab2/main.py
@@ -3,7 +3,7 @@ import numpy as np
 from gen import GARunConfig, genetic_algorithm
 
 
-def fitness_function(chromosome: np.ndarray) -> np.ndarray:
+def fitness_function(chromosome: np.ndarray) -> np.float64:
     return chromosome[0] ** 2 + 2 * chromosome[1] ** 2