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

namespace Fluctio.FluctioSim.Core.Components.MachineLearning.TriggerActions
{
	[AddComponentMenu(Config.PrefixedName+"/Machine Learning/Trigger Actions/Set Reward", Config.ComponentMenuOrder + 60)]
	[DefaultExecutionOrder(31)] // needs to be after all RewardAdder to properly override it
	public class RewardSetter : RewardChanger
	{
		protected override void ChangeReward(float reward) => Agent.UnityAgent.SetReward(reward);

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