&`i)ddlmZddlmZmZddlmZddlmZddl m Z e\Z Z eGddeZ dS) )Optional) OldAPIStackoverride)try_import_torch)Schedule) TensorTypec zeZdZdZ d dedeededeffd Ze e d e d e fd Z xZ S)ExponentialSchedulezExponential decay schedule from `initial_p` to `final_p`. Reduces output over `schedule_timesteps`. After this many time steps always returns `final_p`. N?皙?schedule_timesteps framework initial_p decay_ratect||dksJ||_||_||_dS)aGInitializes a ExponentialSchedule instance. Args: schedule_timesteps: Number of time steps for which to linearly anneal initial_p to final_p. framework: The framework descriptor string, e.g. "tf", "torch", or None. initial_p: Initial output value. decay_rate: The percentage of the original value after 100% of the time has been reached (see formula above). >0.0: The smaller the decay-rate, the stronger the decay. 1.0: No decay at all. )rrN)super__init__r rr)selfr rrr __class__s /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/rllib/utils/schedules/exponential_schedule.pyrzExponentialSchedule.__init__sK( 9---!A%%%%"4"$treturnc|jdkr5tr.t|tjr|}|j|j||jz zzS)z=Returns the result of: initial_p * decay_rate ** (`t`/t_max).torch)rr isinstanceTensorfloatrrr )rrs r_valuezExponentialSchedule._value-sP >W $ $ $:a3N3N $ A~A8O4O PPPr)Nr r )__name__ __module__ __qualname____doc__intrstrrrrrrr __classcell__)rs@rr r s$( %%%C=% %  %%%%%%4XhQ QzQQQQQQQQrr N)typingrray.rllib.utils.annotationsrrray.rllib.utils.frameworkr"ray.rllib.utils.schedules.schedulerray.rllib.utils.typingrr_r rrr.s========666666777777------    q &Q&Q&Q&Q&Q(&Q&Q &Q&Q&Qr