use_jax in AnalysisLens

Jammy2211 · Jammy2211 · commit bc21c238367d · 2025-12-14T18:36:17.000Z
diff --git a/autolens/point/model/analysis.py b/autolens/point/model/analysis.py
@@ -63,7 +63,7 @@ def __init__(
         """
         super().__init__(cosmology=cosmology, use_jax=use_jax)
 
-        AnalysisLens.__init__(self=self, cosmology=cosmology)
+        AnalysisLens.__init__(self=self, cosmology=cosmology, use_jax=use_jax)
 
         self.dataset = dataset
 
diff --git a/autolens/quantity/model/analysis.py b/autolens/quantity/model/analysis.py
@@ -54,7 +54,7 @@ def __init__(
             dataset=dataset, func_str=func_str, cosmology=cosmology, use_jax=use_jax
         )
 
-        AnalysisLens.__init__(self=self, cosmology=cosmology)
+        AnalysisLens.__init__(self=self, cosmology=cosmology, use_jax=use_jax)
 
     def fit_quantity_for_instance(self, instance: af.ModelInstance) -> FitQuantity:
         """

Original file line number	Diff line number	Diff line change
`@@ -54,7 +54,7 @@ def __init__(`
`54`	`54`	`dataset=dataset, func_str=func_str, cosmology=cosmology, use_jax=use_jax`
`55`	`55`	`)`
`56`	`56`
`57`		`- AnalysisLens.__init__(self=self, cosmology=cosmology)`
	`57`	`+ AnalysisLens.__init__(self=self, cosmology=cosmology, use_jax=use_jax)`
`58`	`58`
`59`	`59`	`def fit_quantity_for_instance(self, instance: af.ModelInstance) -> FitQuantity:`
`60`	`60`	`"""`