reformating

BindsNET · May 19, 2021 · 97b5350 · 97b5350
1 parent 159cabe
commit 97b5350
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/bindsnet/learning/learning.py b/bindsnet/learning/learning.py
@@ -552,13 +552,17 @@ def _connection_update(self, **kwargs) -> None:
         # Initialize eligibility, P^+, and P^-.
         if not hasattr(self, "p_plus"):
             self.p_plus = torch.zeros(
-                #batch_size, *self.source.shape, device=self.source.s.device
-                batch_size, self.source.n, device=self.source.s.device
+                # batch_size, *self.source.shape, device=self.source.s.device
+                batch_size,
+                self.source.n,
+                device=self.source.s.device,
             )
         if not hasattr(self, "p_minus"):
             self.p_minus = torch.zeros(
                 # batch_size, *self.target.shape, device=self.target.s.device
-                batch_size, self.target.n, device=self.target.s.device
+                batch_size,
+                self.target.n,
+                device=self.target.s.device,
             )
         if not hasattr(self, "eligibility"):
             self.eligibility = torch.zeros(