Merge remote-tracking branch 'origin/main'

jameschapman19 · Oct 26, 2023 · b52751d · b52751d
2 parents 4201b4f + cb47e7c
commit b52751d
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/cca_zoo/_base.py b/cca_zoo/_base.py
@@ -257,7 +257,6 @@ def canonical_loadings_(
                 ]
         return self._canonical_loadings
 
-
     def loadings_(self) -> List[np.ndarray]:
         """
         Compute and return loadings_ for each view. These are cached for performance optimization.

diff --git a/cca_zoo/datasets/simulated.py b/cca_zoo/datasets/simulated.py
@@ -125,9 +125,16 @@ class JointData(BaseData):
     Class for generating simulated data for a linear model with multiple representations.
     """
 
-    def __init__(self, view_features: List[int], latent_dims: int = 1, view_sparsity: Union[List[float], float] = None,
-                 correlation: Union[List[float], float] = 0.99, structure: str = "random",
-                 positive: Union[bool, List[bool]] = False, random_state: Union[int, np.random.RandomState] = None):
+    def __init__(
+        self,
+        view_features: List[int],
+        latent_dims: int = 1,
+        view_sparsity: Union[List[float], float] = None,
+        correlation: Union[List[float], float] = 0.99,
+        structure: str = "random",
+        positive: Union[bool, List[bool]] = False,
+        random_state: Union[int, np.random.RandomState] = None,
+    ):
         self.view_features = view_features
         self.latent_dims = latent_dims
         self.random_state = check_random_state(random_state)