PyPI - CUQIpy - Versions diffs - 0.8.0.post0.dev89__tar.gz → 0.8.0.post0.dev92__tar.gz - Mend

@@ -589,6 +589,31 @@ def test_beta(): #TODO. Make more tests
     FD_gradient = cuqi.utilities.approx_gradient(BD.logpdf, x, epsilon=0.000000001)
     assert np.allclose(BD.gradient(x),FD_gradient,rtol=1e-3) or (np.all(np.isnan(FD_gradient)) and np.all(np.isnan(BD.gradient(x))))
+# Fixture for beta distribution where beta is a likelihood
+@pytest.fixture
+def beta_likelihood():
+    # simple forward model
+    A = cuqi.model.Model(
+        lambda x: x**2,
+        range_geometry=1,
+        domain_geometry=1,
+        gradient=lambda direction, wrt: 2*wrt*direction)
+    # set a gaussian prior
+    x = cuqi.distribution.Gaussian(0, 1)
+    # Beta data distribution
+    y = cuqi.distribution.Beta(A(x),1)
+    # set the observed data
+    y=y(y=0.5)
+    return y
+def test_gradient_for_Beta_as_likelihood_raises_error(beta_likelihood):
+    """Test computing the gradient of the Beta distribution as a likelihood
+    raises a NotImplementedError"""
+    with pytest.raises(NotImplementedError,
+                       match=r"Gradient is not implemented for CUQI Beta."):
+        beta_likelihood.gradient(1)
 @pytest.mark.parametrize("C",[1, np.ones(5), np.eye(5), sps.eye(5), sps.diags(np.ones(5))])
 def test_Gaussian_Cov_sample(C):

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: CUQIpy
-Version: 0.8.0.post0.dev89
+Version: 0.8.0.post0.dev92
 Summary: Computational Uncertainty Quantification for Inverse problems in Python
 Maintainer-email: "Nicolai A. B. Riis" <nabr@dtu.dk>, "Jakob S. Jørgensen" <jakj@dtu.dk>, "Amal M. Alghamdi" <amaal@dtu.dk>, Chao Zhang <chaz@dtu.dk>
 License:                                  Apache License

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: CUQIpy
-Version: 0.8.0.post0.dev89
+Version: 0.8.0.post0.dev92
 Summary: Computational Uncertainty Quantification for Inverse problems in Python
 Maintainer-email: "Nicolai A. B. Riis" <nabr@dtu.dk>, "Jakob S. Jørgensen" <jakj@dtu.dk>, "Amal M. Alghamdi" <amaal@dtu.dk>, Chao Zhang <chaz@dtu.dk>
 License:                                  Apache License

@@ -8,11 +8,11 @@ import json
 version_json = '''
 {
- "date": "2024-02-07T21:09:32+0100",
+ "date": "2024-02-19T05:04:22+0100",
  "dirty": false,
  "error": null,
- "full-revisionid": "bacee5757cf16e402ca3732feb2aca813f0ccdb4",
- "version": "0.8.0.post0.dev89"
+ "full-revisionid": "f274c4a7b3044d1ba3e9567bb167a1cdbea05283",
+ "version": "0.8.0.post0.dev92"
 }
 '''  # END VERSION_JSON

@@ -69,7 +69,7 @@ class Beta(Distribution):
     def _sample(self, N=1, rng=None):
         return sps.beta.rvs(a=self.alpha, b=self.beta, size=(N,self.dim), random_state=rng).T
-    def _gradient(self, x):
+    def _gradient(self, x, *args, **kwargs):
         #Avoid complicated geometries that change the gradient.
         if not type(self.geometry) in _get_identity_geometries():
             raise NotImplementedError("Gradient not implemented for distribution {} with geometry {}".format(self,self.geometry))

CUQIpy 0.8.0.post0.dev89tar.gz → 0.8.0.post0.dev92tar.gz