Merge remote-tracking branch 'origin/main'

cca_zoo/deep/_discriminative/_dcca.py

@@ -4,6 +4,7 @@
 from cca_zoo.deep._base import BaseDeep
 from cca_zoo.linear._mcca import MCCA
 
+
 class DCCA(BaseDeep):
     """
     A class used to fit a DCCA model.

cca_zoo/utils/check_values.py

@@ -45,6 +45,7 @@ def _check_batch_size(batch_size, latent_dimensions):
             "Objective is unstable when batch size is less than the number of latent dimensions"
         )
 
+
 def check_tsne_support(caller_name):
     try:
         import openTSNE
@@ -54,6 +55,7 @@ def check_tsne_support(caller_name):
             "Please install openTSNE using `pip install openTSNE`"
         )
 
+
 def check_umap_support(caller_name):
     try:
         import umap
@@ -63,6 +65,7 @@ def check_umap_support(caller_name):
             "Please install umap using `pip install umap-learn`"
         )
 
+
 def check_seaborn_support(caller_name):
     try:
         import seaborn
@@ -72,11 +75,12 @@ def check_seaborn_support(caller_name):
             "Please install seaborn using `pip install seaborn`"
         )
 
+
 def check_arviz_support(caller_name):
     try:
         import arviz as az
     except ImportError:
         raise ImportError(
             f"{caller_name} requires arviz. "
             "Please install arviz using `pip install arviz`"
-        )
+        )

cca_zoo/visualisation/__init__.py

@@ -17,5 +17,5 @@
     "CovarianceHeatmapDisplay",
     "TSNEScoreDisplay",
     "UMAPScoreDisplay",
-    "WeightInferenceDisplay"
+    "WeightInferenceDisplay",
 ]

cca_zoo/visualisation/inference.py

@@ -1,5 +1,6 @@
 from cca_zoo.utils.check_values import check_arviz_support
 
+
 class WeightInferenceDisplay:
     """
     Class for displaying inference-related plots.
@@ -76,6 +77,7 @@ def from_mcmc(cls, mcmc, true_features=None):
             An InferenceDisplay instance.
         """
         import arviz as az
+
         idata = az.from_numpyro(mcmc)
         return cls(idata, 2, true_features)
 
@@ -109,4 +111,3 @@ def plot(self):
             plt.tight_layout()
 
         plt.show()
-

docs/source/examples/plot_probabilistic.py

@@ -127,4 +127,3 @@ def sample(self, n):
 
 WeightInferenceDisplay.from_estimator(pcca).plot()
 plt.show()
-