Merge remote-tracking branch 'origin/main'

jameschapman19 · Oct 6, 2023 · a969fce · a969fce
2 parents 6b049ec + dd4c91b
commit a969fce
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/cca_zoo/probabilistic/_pls.py b/cca_zoo/probabilistic/_pls.py
@@ -79,8 +79,8 @@ def _model(self, views):
         )
 
         # Add positive-definite constraint for psi1 and psi2
-        psi1 = jnp.eye(self.n_features_[0])*1e-3
-        psi2 = jnp.eye(self.n_features_[1])*1e-3
+        psi1 = jnp.eye(self.n_features_[0]) * 1e-3
+        psi2 = jnp.eye(self.n_features_[1]) * 1e-3
 
         mu1 = numpyro.param(
             "mu_1",

diff --git a/cca_zoo/probabilistic/_rcca.py b/cca_zoo/probabilistic/_rcca.py
@@ -144,15 +144,15 @@ def _model(self, views):
                 "X1",
                 dist.MultivariateNormal(
                     z @ W1.T + mu1,
-                    covariance_matrix=(1-self.c[0])*psi1,
+                    covariance_matrix=(1 - self.c[0]) * psi1,
                 ),
                 obs=X1,
             )
             numpyro.sample(
                 "X2",
                 dist.MultivariateNormal(
                     z @ W2.T + mu2,
-                    covariance_matrix=(1-self.c[1])*psi2,
+                    covariance_matrix=(1 - self.c[1]) * psi2,
                 ),
                 obs=X2,
             )

diff --git a/test/test_probabilistic.py b/test/test_probabilistic.py
@@ -70,7 +70,6 @@ def test_cca_vs_probabilisticPLS(setup_data):
     ), f"Expected correlation with PLS greater than CCA, got {correlation_pls} and {correlation_cca}"
 
 
-
 def test_cca_vs_probabilisticRidgeCCA(setup_data):
     X, Y, joint = setup_data
     # Initialize models with different regularization parameters