From caad6dc77962d4cbfa7fdc2800b3f0da5dcfbecb Mon Sep 17 00:00:00 2001 From: franksim Date: Thu, 16 Oct 2025 14:15:40 +0000 Subject: [PATCH] assignment-a1: fix --- mmp/a1/tensors.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/mmp/a1/tensors.py b/mmp/a1/tensors.py index 4669751..f96fbac 100644 --- a/mmp/a1/tensors.py +++ b/mmp/a1/tensors.py @@ -2,8 +2,7 @@ import torch def avg_color(img: torch.Tensor): - result = img.mean(dim=(1, 2)).tolist() - return tuple(result) + return img.mean(dim=(1, 2)).tolist() def mask(foreground: torch.Tensor, background: torch.Tensor, mask_tensor: torch.Tensor, threshold: float):