Skip to content

Commit c9a2f3f

Browse files
authored
Fix CategoryCount with multiple labels if some labels are missing (#1838)
1 parent c631d53 commit c9a2f3f

2 files changed

Lines changed: 6 additions & 3 deletions

File tree

src/evidently/metrics/column_statistics.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -326,7 +326,10 @@ def _calculate_value(self, dataset: Dataset):
326326
# only one boolean label is possible here
327327
value = counts[self.metric.categories[0]] # type: ignore[index]
328328
else:
329-
value = counts.loc[self.metric.categories].sum() # type: ignore[index]
329+
value = 0
330+
for cat in self.metric.categories:
331+
if cat in counts:
332+
value += counts[cat] # type: ignore[index]
330333
except KeyError:
331334
value = 0
332335
total = column.data.count()
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
outs:
2-
- md5: a966532c276e44acff2bf09f06631550.dir
3-
size: 7060724
2+
- md5: 72eb75f3d2e9a5c7c49463f577cc61bf.dir
3+
size: 7062684
44
nfiles: 32
55
hash: md5
66
path: visual.spec.ts-snapshots

0 commit comments

Comments
 (0)