RewardShapingContext
python
RewardShapingContext(device: torch.device, metrics: MetricsTracker, observation_shape: tuple[int, ...], observation_dtype: np.dtype[Any], metrics_prefix: Literal['Actor', 'Learner'] = 'Actor') -> NoneRuntime context shared with trainer-owned reward shapers.
Values
device
python
device: torch.devicemetrics
python
metrics: MetricsTrackermetrics_prefix
python
metrics_prefix: Literal['Actor', 'Learner'] = 'Actor'observation_dtype
python
observation_dtype: np.dtype[Any]observation_shape
python
observation_shape: tuple[int, ...]