Tolerance

surrogate.py

@@ -39,6 +39,7 @@ PARAM_BOUNDS = [
     {"name": "train_frac", "type": "range", "bounds": [0.01, 1.0]},
 ]
 
+tol = 1e-6
 
 class Parameterization(BaseModel):
     N: float  # int
@@ -77,7 +78,7 @@ class Parameterization(BaseModel):
 
         if param["type"] == "range":
             min_val, max_val = param["bounds"]
-            if not min_val <= v <= max_val:
+            if not (min_val - tol) <= v <= (max_val + tol):
                 raise ValueError(
                     f"{info.field_name} must be between {min_val} and {max_val}"
                 )
@@ -89,11 +90,11 @@ class Parameterization(BaseModel):
 
     @model_validator(mode="after")
     def check_constraints(self) -> "Parameterization":
-        if self.betas1 > self.betas2:
+        if (self.betas1 - tol) > (self.betas2 + tol):
             raise ValueError(
                 f"Received betas1={self.betas1} which should be less than betas2={self.betas2}"
             )
-        if self.emb_scaler + self.pos_scaler > 1.0:
+        if self.emb_scaler + self.pos_scaler - tol > 1.0:
             raise ValueError(
                 f"Received emb_scaler={self.emb_scaler} and pos_scaler={self.pos_scaler} which should sum to less than or equal to 1.0"  # noqa: E501
             )