Update ssim.py

Author: rittik9

src/torchmetrics/functional/image/ssim.py

@@ -111,7 +111,7 @@ def _ssim_update(
         raise ValueError(f"Expected `sigma` to have positive number. Got {sigma}.")
 
     if data_range is None:
-        data_range = max(preds.max() - preds.min(), target.max() - target.min())
+        data_range = max(preds.max() - preds.min(), target.max() - target.min()).item()
     elif isinstance(data_range, tuple):
         preds = torch.clamp(preds, min=data_range[0], max=data_range[1])
         target = torch.clamp(target, min=data_range[0], max=data_range[1])