feat(ai): add io

src/ai/__init__.py

@@ -1,4 +1,5 @@
+from .io import load_genome, save_genome
 from .log import log
 from .training import train
 
-__all__ = ["log", "train"]
+__all__ = ["log", "train", "load_genome", "save_genome"]

src/ai/io.py

@@ -0,0 +1,15 @@
+import pickle
+from pathlib import Path
+
+import neat
+from utils import BASE_PATH
+
+
+def load_genome() -> neat.DefaultGenome:
+    with open(BASE_PATH / "winner.pkl", "rb") as f:
+        return pickle.load(f)
+
+
+def save_genome(genome: neat.DefaultGenome) -> None:
+    with open(BASE_PATH / "winner.pkl", "wb") as f:
+        pickle.dump(genome, f)