| 51 | |
| 52 | @dataclass |
| 53 | class MockResponse: |
| 54 | response_text: str |
| 55 | reward: float = 0.0 |
| 56 | metrics: Optional[Dict[str, float]] = None |
| 57 | info: Optional[Dict] = None |
| 58 | unique_id: Optional[str] = "0" |
| 59 | tokens: Optional[Tensor] = Tensor([0, 0]) |
| 60 | prompt_length: int = 1 |
| 61 | eid: EID = field(default_factory=EID) |
| 62 | truncate_status: str = "not_truncated" |
| 63 | action_mask: Optional[Tensor] = None |
| 64 | |
| 65 | |
| 66 | class DummyWorkflow(Workflow): |
no outgoing calls