reformat using Black

ChEB-AI · Jan 10, 2025 · 3b17487 · 3b17487
1 parent 976f2b8
commit 3b17487
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/chebai/models/ffn.py b/chebai/models/ffn.py
@@ -4,11 +4,18 @@
 import torch
 from torch import Tensor
 
+
 class FFN(ChebaiBaseNet):
 
     NAME = "FFN"
 
-    def __init__(self, input_size: int = 1000, num_hidden_layers: int = 3, hidden_size: int = 128, **kwargs):
+    def __init__(
+        self,
+        input_size: int = 1000,
+        num_hidden_layers: int = 3,
+        hidden_size: int = 128,
+        **kwargs
+    ):
         super().__init__(**kwargs)
 
         self.layers = torch.nn.ModuleList()
@@ -52,4 +59,3 @@ def forward(self, data, **kwargs):
         for layer in self.layers:
             x = torch.relu(layer(x))
         return {"logits": x}
-