ai-snake-lab 0.4.4__py3-none-any.whl → 0.4.6__py3-none-any.whl

ai_snake_lab/ai/EpsilonAlgo.py

@@ -16,9 +16,8 @@ can also be passed in when invoking the main asg.py front end.
 
 import random
 from random import randint
-import os, sys
 
-from constants.DEpsilon import DEpsilon
+from ai_snake_lab.constants.DEpsilon import DEpsilon
 
 
 class EpsilonAlgo:

ai_snake_lab/ai/ReplayMemory.py

@@ -12,12 +12,12 @@ This file contains the ReplayMemory class.
 
 import os
 from collections import deque
-import random, sys
+import random
 import sqlite3, pickle
 
-from constants.DReplayMemory import MEM_TYPE
-from constants.DFile import DFile
-from constants.DDir import DDir
+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
 
 
 class ReplayMemory:

ai_snake_lab/game/GameBoard.py

@@ -18,7 +18,7 @@ from textual.reactive import var
 from rich.segment import Segment
 from rich.style import Style
 
-from game.GameElements import Direction
+from ai_snake_lab.game.GameElements import Direction
 
 emptyA = "#111111"
 emptyB = "#000000"

ai_snake_lab/game/SnakeGame.py

@@ -14,8 +14,8 @@ import numpy as np
 
 from textual.geometry import Offset
 
-from game.GameBoard import GameBoard
-from game.GameElements import Direction
+from ai_snake_lab.game.GameBoard import GameBoard
+from ai_snake_lab.game.GameElements import Direction
 
 # Maximum number of moves. This is multiplied by the length of the snake. The game
 # ends if game moves > MAX_MOVES * length-of-snake. This avoids enless AI looping behavior.
@@ -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/{AISim.py → 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.4.dist-info → ai_snake_lab-0.4.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ai-snake-lab
-Version: 0.4.4
+Version: 0.4.6
 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.4.dist-info → ai_snake_lab-0.4.6.dist-info}/RECORD

@@ -1,8 +1,7 @@
-ai_snake_lab/AISim.py,sha256=GiqzvU0LAvjjmjP_0JDeak7DAF0R-pEzrPgVpCN-gs8,15148
 ai_snake_lab/ai/AIAgent.py,sha256=p_O7zC15s_tBcI7oqu60X2-3AbrzAWlGXWkrLVC4UsQ,3005
 ai_snake_lab/ai/AITrainer.py,sha256=ssX6B03yZLEKhNCJv9D83iFAEEhU3_XbW4sA0z5QMRM,3304
-ai_snake_lab/ai/EpsilonAlgo.py,sha256=iNVXOqlyjh5vlFePpksR0PTT9M5FDDwJi7P6pYegSrA,2144
-ai_snake_lab/ai/ReplayMemory.py,sha256=eGdszKKeYuvppmmkA9AtypLRbGZQGZO-uCSHb8qcesw,3986
+ai_snake_lab/ai/EpsilonAlgo.py,sha256=Q2U_Ow28ZRQn1hlLybWEhB2Gn101x99mrQWupAooRjk,2142
+ai_snake_lab/ai/ReplayMemory.py,sha256=dtF6CkYWBvjqnKPdNA_0Dgr1DWQYAE-KMpB9IIxkoCk,4020
 ai_snake_lab/ai/models/ModelL.py,sha256=hK7MoPyIF1_K43R_9xW_MYuaweQGo9qSMUIVrHDQZnQ,1249
 ai_snake_lab/ai/models/ModelRNN.py,sha256=Ky63BUqJEmwe1PbM4gtjvqd7SmSy_2XA1n-yu8DNexI,1104
 ai_snake_lab/constants/DDb4EPlot.py,sha256=2g8SdDVZWrTVSTw-rpO84OrNfl7ToBYtf6J7sYB7q8w,442
@@ -18,14 +17,15 @@ ai_snake_lab/constants/DModelLRNN.py,sha256=nyPYa1-8JB7KERWEJM5jBFH3PmISNcr9JPi-
 ai_snake_lab/constants/DReplayMemory.py,sha256=m0SSxFL6v2mVXVxf0lJTJzKfXguxrnsnrJph0v-2lX8,589
 ai_snake_lab/constants/DSim.py,sha256=ULi_e9g2eBPPZv_swQQoagKLT8dL3gJTwlQihkQk8RY,510
 ai_snake_lab/constants/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_snake_lab/game/GameBoard.py,sha256=VjjNXnZZJ_lTo4bcxflB9TWxi792Do7yMnvL3umrMCc,8452
+ai_snake_lab/game/GameBoard.py,sha256=iNgain-8YQg2Ky0GkmGX-ViVmL8P7zPiZYU6Qwrp9Cc,8465
 ai_snake_lab/game/GameElements.py,sha256=9vcNH2YWAzYyBZosvj1DzO6ZmUCUMUd8MaIhORWQ8go,470
-ai_snake_lab/game/SnakeGame.py,sha256=_gyxhLPggPF7B2XvmWLgWP2ywnYWX7QjmwLmi21QuIw,6476
+ai_snake_lab/game/SnakeGame.py,sha256=ADJSVD-bXBruSHf0rzQFh2P9hPWUAnCkQ95B-RFk1vI,6506
+ai_snake_lab/ui/AISim.py,sha256=_XUJtiQ_CMWB7beiXq3o9wgmIL-3QGwjHxy8KbSmd_c,15173
 ai_snake_lab/ui/Db4EPlot.py,sha256=pcEb0ydXNX2wL0EFBSrqhIoTEhryk4GD0Ua8FFEaZHY,5352
 ai_snake_lab/utils/AISim.tcss,sha256=XnhqjuxtOiNQHDyXFrbAYSlCw-6ch4e1o52nbytoeF0,2118
 ai_snake_lab/utils/ConstGroup.py,sha256=ZYyQxFd9PudBUZmc_NsNvWCp___utOe1MptqD3eyVH8,1174
-ai_snake_lab-0.4.4.dist-info/METADATA,sha256=dNvC3Ia4TfGeL5L4N6rS2snjbydNdT_ZCo1XGyYGSMU,3689
-ai_snake_lab-0.4.4.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-ai_snake_lab-0.4.4.dist-info/entry_points.txt,sha256=VXWDqigE8D35J7scdqjR0EVvOjtP1mm0uS76xMnvodo,56
-ai_snake_lab-0.4.4.dist-info/licenses/LICENSE,sha256=f-FHFu0xzHH8O_mvKTw2jUZKhTpw6obpmVOI9rnpKeU,35151
-ai_snake_lab-0.4.4.dist-info/RECORD,,
+ai_snake_lab-0.4.6.dist-info/METADATA,sha256=shfnaQS9q3tefXcTuISZbAJRVCGUQ79ivMNrHgDQkKE,3689
+ai_snake_lab-0.4.6.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+ai_snake_lab-0.4.6.dist-info/entry_points.txt,sha256=ThFx-0yPF5pdwOXUbEo95FEXLUUbOcjfhgMv67PHuIw,59
+ai_snake_lab-0.4.6.dist-info/licenses/LICENSE,sha256=f-FHFu0xzHH8O_mvKTw2jUZKhTpw6obpmVOI9rnpKeU,35151
+ai_snake_lab-0.4.6.dist-info/RECORD,,

ai_snake_lab-0.4.6.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+ai-snake-lab=ai_snake_lab.ui.AISim:main
+

ai_snake_lab-0.4.4.dist-info/entry_points.txt

@@ -1,3 +0,0 @@
-[console_scripts]
-ai-snake-lab=ai_snake_lab.AISim:main
-