junaid0600 commited on
Commit
cb24840
·
verified ·
1 Parent(s): 2f13d79

Update env/models.py

Browse files
Files changed (1) hide show
  1. env/models.py +3 -3
env/models.py CHANGED
@@ -111,7 +111,7 @@ class Reward(BaseModel):
111
  @field_validator("score")
112
  @classmethod
113
  def clamp_score(cls, v):
114
- return max(0.001, min(0.999, round(v, 4)))
115
 
116
  model_config = {"json_schema_extra": {
117
  "example": {
@@ -195,7 +195,7 @@ class BaselineResult(BaseModel):
195
  @field_validator("score")
196
  @classmethod
197
  def clamp_score(cls, v):
198
- return max(0.001, min(0.999, round(float(v), 4)))
199
 
200
  class BaselineResponse(BaseModel):
201
  results: list[BaselineResult]
@@ -215,7 +215,7 @@ class GraderResponse(BaseModel):
215
  @field_validator("score")
216
  @classmethod
217
  def clamp_score(cls, v):
218
- return max(0.001, min(0.999, round(float(v), 4)))
219
 
220
  model_config = {"json_schema_extra": {
221
  "example": {
 
111
  @field_validator("score")
112
  @classmethod
113
  def clamp_score(cls, v):
114
+ return max(0.0, min(1.0, round(v, 4)))
115
 
116
  model_config = {"json_schema_extra": {
117
  "example": {
 
195
  @field_validator("score")
196
  @classmethod
197
  def clamp_score(cls, v):
198
+ return max(0.0, min(1.0, round(float(v), 4)))
199
 
200
  class BaselineResponse(BaseModel):
201
  results: list[BaselineResult]
 
215
  @field_validator("score")
216
  @classmethod
217
  def clamp_score(cls, v):
218
+ return max(0.0, min(1.0, round(float(v), 4)))
219
 
220
  model_config = {"json_schema_extra": {
221
  "example": {