Merge remote-tracking branch 'origin/main'

jameschapman19 · Oct 26, 2023 · a0653be · a0653be
2 parents 44c92f6 + c14e8cb
commit a0653be
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/cca_zoo/deep/architectures.py b/cca_zoo/deep/architectures.py
@@ -110,9 +110,7 @@ def __init__(
                 nn.Dropout(p=dropout), nn.Linear(linear_input_size, latent_dimensions)
             )
 
-    def _build_conv_layers(
-        self, channels, kernel_sizes, strides, paddings, activation
-    ):
+    def _build_conv_layers(self, channels, kernel_sizes, strides, paddings, activation):
         layers = []
         current_channels = 1
         for idx in range(len(channels)):

diff --git a/cca_zoo/linear/_gradient/_ey.py b/cca_zoo/linear/_gradient/_ey.py
@@ -56,9 +56,13 @@ def validation_step(self, batch, batch_idx):
         return loss["objective"]
 
     def get_dataset(self, views: Iterable[np.ndarray], validation_views=None):
-        dataset = DoubleNumpyDataset(views, batch_size=self.batch_size, random_state=self.random_state)
+        dataset = DoubleNumpyDataset(
+            views, batch_size=self.batch_size, random_state=self.random_state
+        )
         if validation_views is not None:
-            val_dataset = DoubleNumpyDataset(validation_views, self.batch_size, self.random_state)
+            val_dataset = DoubleNumpyDataset(
+                validation_views, self.batch_size, self.random_state
+            )
         else:
             val_dataset = None
         return dataset, val_dataset