Updated enemy detection

entities/player.py

@@ -63,6 +63,7 @@ class Player(pygame.sprite.Sprite):
                     no_load=False):
 
         self.get_current_state()
+        self.num_features ==
         self.agent = Agent(
             input_dims=len(self.state_features),
             n_actions=len(self._input.possible_actions),
@@ -173,7 +174,7 @@ class Player(pygame.sprite.Sprite):
 
         enemy_states = []
 
-        for distance, direction, enemy in sorted_distances[:5]:
+        for distance, direction, enemy in self.distance_direction_from_enemy:
             enemy_states.extend([
                 np.exp(-distance),
                 direction[0],

pneuma.py

@@ -223,6 +223,8 @@ if __name__ == "__main__":
                 print(f"\nScore this round for player {player.player_id}:\
                       {player.stats.exp}")
 
+
+
     # End of training session
     print("End of episodes.\
         \nExiting game...")