From e7f1c2f884554c58cffc8fe1d8b3b3f32775d0bd Mon Sep 17 00:00:00 2001
From: Vasilis Valatsos <vasilvalat@gmail.com>
Date: Fri, 24 Nov 2023 19:33:20 +0100
Subject: [PATCH] Updated single-agent.py

---
 .single-agent.py.kate-swp | Bin 0 -> 325 bytes
 single-agent.py           |   6 +++---
 2 files changed, 3 insertions(+), 3 deletions(-)
 create mode 100644 .single-agent.py.kate-swp

diff --git a/.single-agent.py.kate-swp b/.single-agent.py.kate-swp
new file mode 100644
index 0000000000000000000000000000000000000000..9939bf4ce02e424d36a64e5ee251f33a187c0dd3
GIT binary patch
literal 325
zcmZQzU=Z?7EJ;-eE>A2_aLdd|RWQ;sU|?Vn+46UP$iAryt94?z17;U#RagHF-V_`J
zlo1DFP9WxTg>yMTLI^H9kjsJKvH`j52res-%ZA{h>D2<tYC~jIflOr}RzYx;Ktc$v
zB9N<u;3@#QiU_VekgI^;$^p6Z2(B!UD~I690J*Xdt~!tj0;ZurLc=u}M5#lh%s}Em
NU<ML1he(;Z0sz+MC<y=n

literal 0
HcmV?d00001

diff --git a/single-agent.py b/single-agent.py
index aec3e7a..b4fdbf5 100644
--- a/single-agent.py
+++ b/single-agent.py
@@ -47,7 +47,6 @@ for i in tqdm(range(n_episodes)):
 
             for player in game.level.player_sprites:
                 if player.is_dead():
-                    agent_list[0] = player.agent
                     player.kill()
 
             # if (j == game_len-1 or game.level.done) and game.level.enemy_sprites != []:
@@ -55,13 +54,14 @@ for i in tqdm(range(n_episodes)):
             #         for enemy in game.level.enemy_sprites:
             #             player.stats.exp *= .95
 
+    agent_list[0] = game.level.player_sprites[0].agent
+
     for player in game.level.player_sprites:
-        if not player.is_dead():
-            agent_list[0] = player.agent
         exp_points = player.stats.exp
         score_history[player.player_id][i] = exp_points
         avg_score[player.player_id] = np.mean(
             score_history[player.player_id])
+
     if np.mean(avg_score) > np.mean(best_score):
         best_score = avg_score
         print("Saving models for agent...")