ai-snake-lab 0.4.5__tar.gz → 0.4.7__tar.gz

{ai_snake_lab-0.4.5 → ai_snake_lab-0.4.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-snake-lab
-Version: 0.4.5
+Version: 0.4.7
 Summary: Interactive reinforcement learning sandbox for experimenting with AI agents in a classic Snake Game environment.
 License: GPL-3.0
 License-File: LICENSE

{ai_snake_lab-0.4.5 → ai_snake_lab-0.4.7}/ai_snake_lab/ai/ReplayMemory.py

@@ -14,10 +14,11 @@ import os
 from collections import deque
 import random
 import sqlite3, pickle
+import tempfile
+import shutil
 
 from ai_snake_lab.constants.DReplayMemory import MEM_TYPE
-from ai_snake_lab.constants.DFile import DFile
-from ai_snake_lab.constants.DDir import DDir
+from ai_snake_lab.constants.DDef import DDef
 
 
 class ReplayMemory:
@@ -31,11 +32,6 @@ class ReplayMemory:
         self.max_states = 15000
         self.max_shuffle_games = 40
         self.max_games = 500
-        self.db_file = os.path.join(DDir.AI_SNAKE_LAB, DDir.DB, DFile.REPLAY_DB)
-
-        # Delete the replay memory file, if it exists
-        if os.path.exists(self.db_file):
-            os.remove(self.db_file)
 
         if self._mem_type == MEM_TYPE.SHUFFLE:
             # States are stored in a deque and a random sample will be returned
@@ -46,13 +42,33 @@ class ReplayMemory:
             # A complete game will be returned
             self.cur_memory = []
 
+        # Get a temporary directory for the DB file
+        self._tmpfile = tempfile.NamedTemporaryFile(suffix=DDef.DOT_DB, delete=False)
+        self.db_file = self._tmpfile.name
+
         # Connect to SQLite
         self.conn = sqlite3.connect(self.db_file, check_same_thread=False)
+
+        # Get a cursor
         self.cursor = self.conn.cursor()
+
+        # We don't need the file handle anymore
+        self._tmpfile.close()
+
+        # Intialize the schema
         self.init_db()
 
-    def __len__(self):
-        return len(self.memories)
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.close()
+
+    def __del__(self):
+        try:
+            self.close()
+        except Exception:
+            pass  # avoid errors on interpreter shutdown
 
     def append(self, transition):
         """Add a transition to the current game."""
@@ -75,7 +91,12 @@ class ReplayMemory:
 
     def close(self):
         """Close the database connection."""
-        self.conn.close()
+        if getattr(self, "conn", None):
+            self.conn.close()
+            self.conn = None
+        if getattr(self, "db_file", None) and os.path.exists(self.db_file):
+            os.remove(self.db_file)
+            self.db_file = None
 
     def get_random_game(self):
         """Return a random full game from the database."""

{ai_snake_lab-0.4.5 → ai_snake_lab-0.4.7}/ai_snake_lab/constants/DDef.py

@@ -15,4 +15,5 @@ class DDef(ConstGroup):
     """Defaults"""
 
     APP_TITLE: str = "AI Snake Game Lab"
+    DOT_DB: str = ".db"
     MOVE_DELAY: float = 0.0

{ai_snake_lab-0.4.5 → ai_snake_lab-0.4.7}/ai_snake_lab/constants/DDir.py

@@ -15,5 +15,4 @@ class DDir(ConstGroup):
     """Directories"""
 
     AI_SNAKE_LAB: str = "ai_snake_lab"
-    DB: str = "db"
     UTILS: str = "utils"

{ai_snake_lab-0.4.5 → ai_snake_lab-0.4.7}/ai_snake_lab/constants/DFile.py

@@ -15,4 +15,3 @@ class DFile(ConstGroup):
     """Files"""
 
     CSS_FILE: str = "AISim.tcss"
-    REPLAY_DB: str = "replay_mem.db"

{ai_snake_lab-0.4.5 → ai_snake_lab-0.4.7}/ai_snake_lab/game/SnakeGame.py

@@ -129,12 +129,6 @@ class SnakeGame:
             game_over = True
             reward = -10
 
-        # Set a negative reward if the snake head is adjacent to the snake body.
-        # This is to discourage snake collisions.
-        for segment in self.snake[1:]:
-            if abs(self.head.x - segment.x) < 2 and abs(self.head.y - segment.y) < 2:
-                reward -= -1
-
         if game_over == True:
             # Game is over: Snake or wall collision or exceeded max moves
             self.game_reward += reward
@@ -159,6 +153,12 @@ class SnakeGame:
             reward -= 2
         self.distance_to_food = cur_distance
 
+        ## 6. Set a negative reward if the snake head is adjacent to the snake body.
+        # This is to discourage snake collisions.
+        for segment in self.snake[1:]:
+            if abs(self.head.x - segment.x) < 2 and abs(self.head.y - segment.y) < 2:
+                reward -= -2
+
         self.game_reward += reward
         self.game_board.update_snake(snake=self.snake, direction=self.direction)
         self.game_board.update_food(food=self.food)

{ai_snake_lab-0.4.5/ai_snake_lab → ai_snake_lab-0.4.7/ai_snake_lab/ui}/AISim.py

@@ -426,6 +426,10 @@ def minutes_to_uptime(seconds: int):
         return f"{seconds}s"
 
 
-if __name__ == "__main__":
+def main():
     app = AISim()
     app.run()
+
+
+if __name__ == "__main__":
+    main()

{ai_snake_lab-0.4.5 → ai_snake_lab-0.4.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "ai-snake-lab"
-version = "0.4.5"
+version = "0.4.7"
 description = "Interactive reinforcement learning sandbox for experimenting with AI agents in a classic Snake Game environment."
 authors = [{ name = "Nadim-Daniel Ghaznavi", email = "nghaznavi@gmail.com" }]
 license = { text = "GPL-3.0" }
@@ -38,7 +38,7 @@ dependencies = [
 ]
 
 [project.scripts]
-ai-snake-lab = "ai_snake_lab.AISim:main"
+ai-snake-lab = "ai_snake_lab.ui.AISim:main"
 
 [project.urls]
 "Bug Tracker" = "https://github.com/NadimGhaznavi/ai_snake_lab/issues"