From dee490891a4fd2972d5bd55851d74eef292d954d Mon Sep 17 00:00:00 2001
From: Emil Shanaty <65846814+emil28092005@users.noreply.github.com>
Date: Mon, 7 Apr 2025 18:27:12 +0300
Subject: [PATCH] Update features

---
 main.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/main.py b/main.py
index 9dd6219..9edb7e8 100644
--- a/main.py
+++ b/main.py
@@ -33,10 +33,11 @@ spawner = Spawner("images/Barrier.png")  # Initialize spawner
 last_barrier = barriers[0]
 
 def get_features(horse):
-    features = [last_barrier.rect.topleft, last_barrier.rect.bottomleft,
-               last_barrier.rect.bottomright, last_barrier.rect.topright, 
-               horse.rect.topleft, horse.rect.bottomleft, horse.rect.bottomright, horse.rect.topright,
-               HEIGHT]
+    features = [last_barrier.rect.topleft[0], last_barrier.rect.topleft[1],
+               last_barrier.rect.bottomright[0], last_barrier.rect.bottomright[1], 
+               horse.rect.topleft[0], horse.rect.topleft[1],
+               horse.rect.bottomright[0], horse.rect.bottomright[1],
+               HEIGHT, BARRIER_SPEED]
     return features 
 
 UI.add_horses(horses)  # Add horses to UI