From 4a985b7e0130a1bf142feb9f7be3885a5c7ba866 Mon Sep 17 00:00:00 2001
From: Kristofers Solo <kristians.cagulis@pm.me>
Date: Fri, 5 Jan 2024 17:13:03 +0200
Subject: [PATCH] refactor(ai): move `fitness.py`

---
 src/ai/fitness/__init__.py      | 3 +++
 src/ai/{ => fitness}/fitness.py | 6 ------
 2 files changed, 3 insertions(+), 6 deletions(-)
 create mode 100644 src/ai/fitness/__init__.py
 rename src/ai/{ => fitness}/fitness.py (94%)

diff --git a/src/ai/fitness/__init__.py b/src/ai/fitness/__init__.py
new file mode 100644
index 0000000..2e9e82a
--- /dev/null
+++ b/src/ai/fitness/__init__.py
@@ -0,0 +1,3 @@
+from .fitness import calculate_fitness
+
+__all__ = ["calculate_fitness"]
diff --git a/src/ai/fitness.py b/src/ai/fitness/fitness.py
similarity index 94%
rename from src/ai/fitness.py
rename to src/ai/fitness/fitness.py
index b2ce090..dd8b6ef 100644
--- a/src/ai/fitness.py
+++ b/src/ai/fitness/fitness.py
@@ -5,8 +5,6 @@ import numpy as np
 from game import Game
 from utils import CONFIG
 
-from .log import log
-
 
 def calculate_fitness(game: Game) -> float:
     field = np.where(game.field != None, 1, 0)
@@ -15,10 +13,6 @@ def calculate_fitness(game: Game) -> float:
     return fitness
 
 
-def get_peaks(field: np.ndarray) -> int:
-    pass
-
-
 def get_row_transitions(field: np.ndarray) -> int:
     pass