Method __init__

tests/test_stateless_executor.py:51–54

(self, rewards: torch.Tensor)

class FakeRewardModel(torch.nn.Module):
    def __init__(self, rewards: torch.Tensor):
        super().__init__()
        self.register_buffer("fixed_rewards", rewards)
        self.use_cache_calls: list[bool | None] = []

    def forward(self, input_ids, attention_mask=None, use_cache=None):
        del attention_mask