using Fluctio.FluctioSim.Common.Configuration;
using UnityEngine;

namespace Fluctio.FluctioSim.Core.Components.MachineLearning.TriggerActions
{
	[AddComponentMenu(Config.PrefixedName+"/Machine Learning/Trigger Actions/Add Reward", Config.ComponentMenuOrder + 50)]
	[DefaultExecutionOrder(30)] // after some initialization and after randomization
	public class RewardAdder : RewardChanger
	{
		protected override void ChangeReward(float reward) => Agent.UnityAgent.AddReward(reward);

		private void Reset()
		{
			scaleByTimestep = true;
		}
	}
}