We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ef03c8c commit f5cbccaCopy full SHA for f5cbcca
autointent/modules/scoring/_catboost/catboost_scorer.py
@@ -1,7 +1,7 @@
1
"""CatBoostScorer class for CatBoost-based classification with switchable encoding."""
2
3
import logging
4
-from enum import StrEnum
+from enum import Enum
5
from typing import Any, cast
6
7
import numpy as np
@@ -19,7 +19,7 @@
19
logger = logging.getLogger(__name__)
20
21
22
-class FeaturesType(StrEnum):
+class FeaturesType(str, Enum):
23
"""Type of features used in CatBoostScorer."""
24
25
TEXT = "text"
0 commit comments