beet_ml 0.0.8

Machine Learning actions built upon beet_flow
1
2
3
4
5
6
7
8
9
10
11

use crate::prelude::*;
use beet_core::prelude::*;


pub fn reward_grid(
	mut query: Query<(&FrozenLakeMap, &GridPos, &mut Reward), Changed<GridPos>>,
) {
	for (map, pos, mut reward) in query.iter_mut() {
		reward.0 = map.position_to_cell(**pos).reward();
	}
}