remove mmd from two moons test

Author: LarsKue

tests/test_two_moons/test_two_moons.py

@@ -13,12 +13,9 @@ def test_compile(approximator, random_samples, jit_compile):
 
 
 def test_fit(approximator, train_dataset, validation_dataset, batch_size):
-    from bayesflow.metrics import MaximumMeanDiscrepancy
-    from bayesflow.networks import PointInferenceNetwork
-
     inference_metrics = []
-    if not isinstance(approximator.inference_network, PointInferenceNetwork):
-        inference_metrics += [MaximumMeanDiscrepancy()]
+    # if not isinstance(approximator.inference_network, PointInferenceNetwork):
+    #     inference_metrics += [MaximumMeanDiscrepancy()]
     approximator.compile(inference_metrics=inference_metrics)
 
     mock_data = train_dataset[0]
@@ -41,8 +38,8 @@ def test_fit(approximator, train_dataset, validation_dataset, batch_size):
 
     # test that metrics are improving
     metric_names = ["loss"]
-    if not isinstance(approximator.inference_network, PointInferenceNetwork):
-        metric_names += ["maximum_mean_discrepancy/inference_maximum_mean_discrepancy"]
+    # if not isinstance(approximator.inference_network, PointInferenceNetwork):
+    #     metric_names += ["maximum_mean_discrepancy/inference_maximum_mean_discrepancy"]
     for metric in metric_names:
         assert metric in untrained_metrics
         assert metric in trained_metrics