add Raises to compute_hypothesis_test for unmatching observed and reference data + add corresponding test cases

Author: thegialeo

bayesflow/diagnostics/metrics/mmd_hypothesis_test.py

@@ -166,7 +166,18 @@ def compute_mmd_hypothesis_test(
         The MMD value between observed and reference data.
     mmd_null : np.ndarray
         A distribution of MMD values under the null hypothesis.
+
+    Raises:
+    ------
+    ValueError
+        - If the shapes of observed and reference data do not match on dimensions besides the first one.
     """
+    if observed_data.shape[1:] != reference_data.shape[1:]:
+        raise ValueError(
+            f"Expected observed and reference data to have the same shape, "
+            f"but got {observed_data.shape[1:]} != {reference_data.shape[1:]}."
+        )
+
     if approximator.summary_network is not None:
         observed_data_tensor: Tensor = convert_to_tensor(observed_data)
         reference_data_tensor: Tensor = convert_to_tensor(reference_data)

tests/test_diagnostics/test_diagnostics_metrics.py

@@ -249,7 +249,7 @@ def test_compute_hypothesis_test_different_distributions(summary_network, monkey
     assert mmd_observed >= np.quantile(mmd_null, 0.68)
 
 
-@pytest.mark.parametrize("summary_network", [lambda data: np.random.rand(data.shape[0], 5), None])
+@pytest.mark.parametrize("summary_network", [lambda data: np.random.rand(data.shape[0], 5)])
 def test_compute_hypothesis_test_mismatched_last_dimension(summary_network, monkeypatch):
     """Test that a ValueError is raised for mismatched last dimensions."""
     observed_data = np.random.rand(10, 5)