a new commit

Signed-off-by: AymanL <40838419+AymanL@users.noreply.github.com>

Author: AymanL

monai/metrics/frd.py

@@ -61,5 +61,9 @@ def get_frd_score(y_pred: torch.Tensor, y: torch.Tensor) -> torch.Tensor:
 
     Returns:
         Scalar tensor containing the Fréchet Radiomics Distance.
+
+    Raises:
+        ValueError: When either tensor has more than 2 dimensions. Inputs must have
+            shape (number of samples, number of features).
     """
     return get_fid_score(y_pred, y)

tests/metrics/test_compute_frd_metric.py

@@ -38,9 +38,11 @@ def test_frd_matches_fid_for_same_features(self):
         fid_score = FIDMetric()(y_pred, y)
         np.testing.assert_allclose(frd_score.cpu().numpy(), fid_score.cpu().numpy(), atol=1e-6)
 
-    def test_input_dimensions(self):
+    def test_rejects_high_dimensional_input(self):
+        """FrechetRadiomicsDistance raises ValueError when inputs have ndimension() > 2."""
+        high_dim = torch.ones([3, 3, 144, 144])
         with self.assertRaises(ValueError):
-            FrechetRadiomicsDistance()(torch.ones([3, 3, 144, 144]), torch.ones([3, 3, 145, 145]))
+            FrechetRadiomicsDistance()(high_dim, high_dim)
 
 
 if __name__ == "__main__":