Merge pull request #214 from ibm-granite/service_updates

relax parameters strictness
ibm-granite · Nov 30, 2024 · fa81367 · fa81367
2 parents ebd4589 + 27f6a3c
commit fa81367
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/services/boilerplate/inference_payloads.py b/services/boilerplate/inference_payloads.py
@@ -101,12 +101,10 @@ class ForecastingMetadataInput(BaseMetadataInput):
 
 
 class BaseParameters(BaseModel):
-    model_config = ConfigDict(extra="forbid", protected_namespaces=())
-
+    model_config = ConfigDict(extra="allow", protected_namespaces=())
 
-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"