(self)
| 341 | with pytest.raises(ValueError, match="shape \\[B\\]"): |
| 342 | score_rewards(outputs, inputs, reward_adapter=lambda _outputs, _inputs: torch.ones(2, 2)) |
| 343 | |
| 344 | with pytest.raises(ValueError, match="batch size"): |
| 345 | score_rewards(outputs, inputs, reward_adapter=lambda _outputs, _inputs: torch.ones(1)) |
| 346 | |
| 347 | |
| 348 | def test_default_reward_adapter_rejects_non_scalar_logits(): |
| 349 | inputs = _inputs() |