fix final unit test

Author: Jammy2211

test_autofit/graphical/global/test_hierarchical.py

@@ -55,7 +55,8 @@ def test_model_info(model):
 2 - 3
     one                                                                         UniformPrior [0], lower_limit = 0.0, upper_limit = 1.0
 factor
-    include_prior_factors                                                       True"""
+    include_prior_factors                                                       True
+    use_jax                                                                     False"""
     )
 
 

test_autofit/mapper/test_array.py

@@ -178,6 +178,9 @@ def log_likelihood_function(self, instance):
 
 
 def test_optimisation():
+
+    import jax.numpy as jnp
+
     array = af.Array(
         shape=(2, 2),
         prior=af.UniformPrior(
@@ -192,4 +195,5 @@ def test_optimisation():
     array[0, 1] = posterior[0, 1]
 
     result = af.DynestyStatic().fit(model=array, analysis=Analysis())
-    assert isinstance(result.instance, np.ndarray)
+
+    assert isinstance(result.instance, jnp.ndarray)