From bb88f753ddad096e8d9f75c6ddacc224ddd680c1 Mon Sep 17 00:00:00 2001 From: Irozuku Date: Wed, 28 Jan 2026 11:37:32 -0300 Subject: [PATCH] fix: remove probability parameter from SVCSchema and set it to True in SVC constructor --- DashAI/back/models/scikit_learn/svc.py | 16 +--------------- 1 file changed, 1 insertion(+), 15 deletions(-) diff --git a/DashAI/back/models/scikit_learn/svc.py b/DashAI/back/models/scikit_learn/svc.py index d97b7a43c..e52cd0601 100644 --- a/DashAI/back/models/scikit_learn/svc.py +++ b/DashAI/back/models/scikit_learn/svc.py @@ -119,21 +119,6 @@ class SVCSchema(BaseSchema): ), alias=MultilingualString(en="max iterations", es="max iteraciones"), ) # type: ignore - probability: schema_field( - bool_field(), - placeholder=True, - description=MultilingualString( - en=( - "The parameter 'probability' indicates whether or not " - "to predict with probabilities." - ), - es=( - "El parĂ¡metro 'probabilidad' indica si " - "se debe predecir o no con probabilidades." - ), - ), - alias=MultilingualString(en="probability", es="probabilidad"), - ) # type: ignore shrinking: schema_field( bool_field(), placeholder=True, @@ -205,4 +190,5 @@ class SVC(TabularClassificationModel, SklearnLikeClassifier, _SVC): CATEGORICAL_ENCODING = CategoricalEncodingStrategy.ONE_HOT def __init__(self, **kwargs): + kwargs["probability"] = True super().__init__(**kwargs)