Skip to content

Commit f5cbcca

Browse files
committed
fix strenum
1 parent ef03c8c commit f5cbcca

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

autointent/modules/scoring/_catboost/catboost_scorer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
"""CatBoostScorer class for CatBoost-based classification with switchable encoding."""
22

33
import logging
4-
from enum import StrEnum
4+
from enum import Enum
55
from typing import Any, cast
66

77
import numpy as np
@@ -19,7 +19,7 @@
1919
logger = logging.getLogger(__name__)
2020

2121

22-
class FeaturesType(StrEnum):
22+
class FeaturesType(str, Enum):
2323
"""Type of features used in CatBoostScorer."""
2424

2525
TEXT = "text"

0 commit comments

Comments
 (0)