Source code for ray.rllib.utils.schedules.constant_schedule

from ray.rllib.utils.annotations import override
from ray.rllib.utils.framework import try_import_tf
from ray.rllib.utils.schedules.schedule import Schedule

tf1, tf, tfv = try_import_tf()

[docs]class ConstantSchedule(Schedule): """ A Schedule where the value remains constant over time. """ def __init__(self, value, framework): """ Args: value (float): The constant value to return, independently of time. """ super().__init__(framework=framework) self._v = value @override(Schedule) def _value(self, t): return self._v @override(Schedule) def _tf_value_op(self, t): return tf.constant(self._v)