File size: 635 Bytes
de07414 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 | from pydantic import BaseModel
from typing import Optional, Any, Dict, List
class Action(BaseModel):
action_str: str
# Alias for compatibility with the sample inference script
BrowserGymAction = Action
class Observation(BaseModel):
goal: str
result: str
step: int
last_action_error: bool
schema_dump: Optional[str] = None
class Reward(BaseModel):
value: float
reason: Optional[str] = None
class StepResult(BaseModel):
observation: Observation
reward: float
done: bool
info: Dict[str, Any]
class ResetResult(BaseModel):
observation: Observation
info: Dict[str, Any]
|