We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 20cad6d commit 635efe9Copy full SHA for 635efe9
gobbli/inspect/evaluate.py
@@ -174,7 +174,7 @@ def metrics_report(self) -> str:
174
y_true: Union[pd.DataFrame, List[str]] = self.y_true_multilabel
175
y_pred: Union[pd.DataFrame, List[str]] = self.y_pred_multilabel
176
# Since these are indicator dataframes, the "labels" are indices
177
- labels = list(range(len(self.labels)))
+ labels: Union[List[str], List[int]] = list(range(len(self.labels)))
178
else:
179
y_true = self.y_true_multiclass
180
y_pred = self.y_pred_multiclass
0 commit comments