Format code with black

jameschapman19 · Sep 18, 2023 · ed038fe · ed038fe
1 parent 2d96b16
commit ed038fe
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/cca_zoo/deep/_discriminative/_dcca_gha.py b/cca_zoo/deep/_discriminative/_dcca_gha.py
@@ -28,7 +28,7 @@ def loss(self, views, independent_views=None, **kwargs):
         rewards = torch.trace(2 * A)
         if independent_views is None:
             # Hebbian
-            penalties= torch.trace(A.detach() @ B)
+            penalties = torch.trace(A.detach() @ B)
             # penalties = torch.trace(A @ B)
         else:
             independent_z = self(independent_views)

diff --git a/cca_zoo/linear/_gradient/_ey.py b/cca_zoo/linear/_gradient/_ey.py
@@ -78,6 +78,7 @@ def get_dataset(self, views: Iterable[np.ndarray]):
 
 class DoubleNumpyDataset(NumpyDataset):
     random_state = np.random.RandomState(0)
+
     def __getitem__(self, index):
         views = [view[index] for view in self.views]
         independent_index = self.random_state.randint(0, len(self))

diff --git a/cca_zoo/linear/_gradient/_gha.py b/cca_zoo/linear/_gradient/_gha.py
@@ -27,7 +27,7 @@ def loss(self, views, independent_views=None, **kwargs):
         z = self(views)
         # Getting A and B matrices from z
         A, B = self.get_AB(z)
-        rewards= torch.trace(2*A)
+        rewards = torch.trace(2 * A)
         if independent_views is None:
             # Hebbian
             penalties = torch.trace(A.detach() @ B)