Added fruit status

celeste/celeste_ai/celeste.py

@@ -28,6 +28,9 @@ class CelesteState(NamedTuple):
 	# Number of deaths since game start
 	deaths: int
 
+	# If an index is true, we got a strawberry on that stage.
+	berries: list[bool]
+
 	# Distance to next point
 	dist: float
 
@@ -223,6 +226,7 @@ class Celeste:
 				xvel			= float(self._internal_state["vx"]),
 				yvel			= float(self._internal_state["vy"]),
 				deaths			= int(self._internal_state["dc"]),
+				berries			= [x == "t" for x in self._internal_state["fr"][1:]],
 
 				dist			= self._dist,
 				next_point		= self._next_checkpoint_idx,