We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 79871ab commit d4e0d71Copy full SHA for d4e0d71
1 file changed
cobra/preprocessing/categorical_data_processor.py
@@ -71,7 +71,8 @@ def __init__(self,
71
scale_contingency_table: bool = True,
72
forced_categories: dict = {}):
73
74
- # assert model_type in ["classification", "regression"], "define an appropriate model type"
+ if model_type not in ["classification", "regression"]:
75
+ raise ValueError("An unexpected model_type was provided. Valid model_types are either 'classification' or 'regression'.")
76
77
self.model_type = model_type
78
self.regroup = regroup
0 commit comments