Skip to content

Commit aed0987

Browse files
committed
fix typing
1 parent 39f3f80 commit aed0987

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

autointent/modules/scoring/_sklearn/sklearn_scorer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
from typing_extensions import Self
1010

1111
from autointent import Context, Embedder
12-
from autointent.custom_types import LabelType
12+
from autointent.custom_types import ListOfLabels
1313
from autointent.modules.abc import ScoringModule
1414

1515
logger = logging.getLogger(__name__)
@@ -103,7 +103,7 @@ def from_context(
103103
def fit(
104104
self,
105105
utterances: list[str],
106-
labels: list[LabelType],
106+
labels: ListOfLabels,
107107
) -> None:
108108
"""
109109
Train the chosen sklearn classifier.

0 commit comments

Comments
 (0)