Update get_features()

2025-04-06 17:54:56 +03:00
parent ce3af54669
commit 9bedb55c5b
1 changed files with 2 additions and 1 deletions
@@ -35,7 +35,8 @@ 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, 
-               HEIGHT, horse.rect.topleft]
+               horse.rect.topleft, horse.rect.bottomleft, horse.rect.bottomright, horse.rect.topright,
+               HEIGHT]
    return features 

 UI.add_horses(horses)  # Add horses to UI