Skip to content

Commit 872804f

Browse files
authored
fix ImagePair MAE comparison (#7477)
1 parent 0387b88 commit 872804f

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

test/common_utils.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -292,7 +292,8 @@ def compare(self) -> None:
292292
actual, expected = self._equalize_attributes(actual, expected)
293293

294294
if self.mae:
295-
actual, expected = self._promote_for_comparison(actual, expected)
295+
if actual.dtype is torch.uint8:
296+
actual, expected = actual.to(torch.int), expected.to(torch.int)
296297
mae = float(torch.abs(actual - expected).float().mean())
297298
if mae > self.atol:
298299
self._fail(

0 commit comments

Comments
 (0)