subtract LH penalty

Author: Jammy2211

autolens/analysis/positions.py

@@ -191,6 +191,12 @@ def log_likelihood_penalty_from(
 
         penalty = self.log_likelihood_penalty_factor * (max_separation - self.threshold)
 
+        print(max_separation, self.threshold, jax.lax.cond(
+            max_separation > self.threshold,
+            lambda: penalty,
+            lambda: jnp.array(0.0),
+        ))
+
         return jax.lax.cond(
             max_separation > self.threshold,
             lambda: penalty,

autolens/imaging/model/analysis.py

@@ -89,7 +89,7 @@ def log_likelihood_function(self, instance: af.ModelInstance) -> float:
             instance=instance
         )
 
-        return self.fit_from(instance=instance).figure_of_merit + log_likelihood_penalty
+        return self.fit_from(instance=instance).figure_of_merit - log_likelihood_penalty
 
     def fit_from(
         self,

autolens/interferometer/model/analysis.py

@@ -153,7 +153,7 @@ def log_likelihood_function(self, instance):
 
         log_likelihood_penalty = self.log_likelihood_penalty_from(instance=instance)
 
-        return self.fit_from(instance=instance).figure_of_merit + log_likelihood_penalty
+        return self.fit_from(instance=instance).figure_of_merit - log_likelihood_penalty
 
     def fit_from(
         self,