Skip add_loss test for jax backend

Author: han-ol

tests/test_approximators/test_add_loss.py

@@ -23,6 +23,11 @@ def call(self, x, training=False, **kwargs):
 
 
 def test_layer_loss_reported(approximator_using_add_loss, train_dataset, validation_dataset):
+    from bayesflow.approximators.backend_approximators.jax_approximator import JAXApproximator
+
+    if isinstance(approximator_using_add_loss, JAXApproximator):
+        pytest.skip(reason="With JAX backend, the compute_metrics method currently fails to consider self.losses.")
+
     approximator = approximator_using_add_loss
     approximator.compile(optimizer="AdamW")
     num_epochs = 3