Spaces:
Runtime error
Runtime error
File size: 1,980 Bytes
2a33798 |
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 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 |
from pydantic import BaseModel, Field, validator
# Define your desired data structure.
class TwoAction(BaseModel):
action: int = Field(description="the choosed action to perform")
# You can add custom validation logic easily with Pydantic.
@validator('action')
def action_is_valid(cls, field):
if field not in [1, 2]:
raise ValueError("Action is not valid ([1, 2])!")
return field
class ThreeAction(BaseModel):
action: int = Field(description="the choosed action to perform")
# You can add custom validation logic easily with Pydantic.
@validator('action')
def action_is_valid(cls, field):
if field not in [1, 2, 3]:
raise ValueError("Action is not valid ([1, 2, 3])!")
return field
class FourAction(BaseModel):
action: int = Field(description="the choosed action to perform")
# You can add custom validation logic easily with Pydantic.
@validator('action')
def action_is_valid(cls, field):
if field not in [1, 2, 3, 4]:
raise ValueError("Action is not valid ([1, 2, 3, 4])!")
return field
class SixAction(BaseModel):
action: int = Field(description="the choosed action to perform")
# You can add custom validation logic easily with Pydantic.
@validator('action')
def action_is_valid(cls, field):
if field not in [1, 2, 3, 4, 5, 6]:
raise ValueError("Action is not valid ([1, 2, 3, 4, 5, 6])!")
return field
class ContinuousAction(BaseModel):
action: float = Field(description="the choosed action to perform")
# You can add custom validation logic easily with Pydantic.
@validator('action')
def action_is_valid(cls, field):
if not (field >= -1 and field <= 1):
raise ValueError("Action is not valid ([-1,1])!")
return field
PARSERS = {1:ContinuousAction, 2: TwoAction, 3: ThreeAction, 4: FourAction, 6: SixAction}
|