Skip to content

Commit 9f4a001

Browse files
SkafteNickiBorda
authored andcommitted
Hotfix doctest failing on master (#3022)
1 parent f162bde commit 9f4a001

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Diff for: src/torchmetrics/classification/group_fairness.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -307,7 +307,7 @@ def plot(
307307
>>> # Example plotting a single value
308308
>>> from torchmetrics.classification import BinaryFairness
309309
>>> metric = BinaryFairness(2)
310-
>>> metric.update(rand(20), randint(2, (20,)), ones(20).long())
310+
>>> metric.update(rand(50), randint(2, (50,)), ones(50).long())
311311
>>> fig_, ax_ = metric.plot()
312312
313313
.. plot::
@@ -319,7 +319,7 @@ def plot(
319319
>>> metric = BinaryFairness(2)
320320
>>> values = [ ]
321321
>>> for _ in range(10):
322-
... values.append(metric(rand(20), randint(2, (20,) ), ones(20).long()))
322+
... values.append(metric(rand(50), randint(2, (50,) ), ones(50).long()))
323323
>>> fig_, ax_ = metric.plot(values)
324324
325325
"""

0 commit comments

Comments
 (0)