correct max rule count in survival problems

adam-bialy · adam-bialy · commit 12ee4587e574 · 2025-01-02T10:05:58.000+01:00
diff --git a/rulekit/survival.py b/rulekit/survival.py
@@ -42,6 +42,7 @@ class _SurvivalModelsParams(BaseModel):
     complementary_conditions: Optional[bool] = DEFAULT_PARAMS_VALUE[
         "complementary_conditions"
     ]
+    max_rule_count: int = DEFAULT_PARAMS_VALUE["max_rule_count"]
 
 
 class _SurvivalExpertModelParams(_SurvivalModelsParams, ExpertModelParams):

Original file line number	Diff line number	Diff line change
`@@ -42,6 +42,7 @@ class _SurvivalModelsParams(BaseModel):`
`42`	`42`	`complementary_conditions: Optional[bool] = DEFAULT_PARAMS_VALUE[`
`43`	`43`	`"complementary_conditions"`
`44`	`44`	`]`
	`45`	`+ max_rule_count: int = DEFAULT_PARAMS_VALUE["max_rule_count"]`
`45`	`46`
`46`	`47`
`47`	`48`	`class _SurvivalExpertModelParams(_SurvivalModelsParams, ExpertModelParams):`