cactus183/patchbench-dev

1from typing import Any2 3from pydantic import BaseModel, Field4 5 6class PatchBenchObservation(BaseModel):7    task_description: str8    buggy_code: str9    failing_tests: str10    step_number: int11    max_steps: int12    reward: float = 0.013    done: bool = False14    info: dict[str, Any] = Field(default_factory=dict)15 16 17class PatchBenchAction(BaseModel):18    patched_code: str19