relax parameters strictness

ibm-granite · Nov 26, 2024 · 1f00c9a · 1f00c9a
1 parent ebd4589
commit 1f00c9a
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/services/boilerplate/inference_payloads.py b/services/boilerplate/inference_payloads.py
@@ -100,13 +100,10 @@ class ForecastingMetadataInput(BaseMetadataInput):
     )
 
 
-class BaseParameters(BaseModel):
-    model_config = ConfigDict(extra="forbid", protected_namespaces=())
-
+class BaseParameters(BaseModel): ...
 
-class ForecastingParameters(BaseModel):
-    model_config = ConfigDict(extra="forbid", protected_namespaces=())
 
+class ForecastingParameters(BaseParameters):
     prediction_length: Optional[int] = Field(
         description="The prediction length for the forecast."
         " The service will return this many periods beyond the last"