[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

computer_vision/vision_transformer.py

@@ -158,7 +158,10 @@ def add_positional_encoding(
 
 
 def attention_mechanism(
-    query: np.ndarray, key: np.ndarray, value: np.ndarray, mask: Optional[np.ndarray] = None
+    query: np.ndarray,
+    key: np.ndarray,
+    value: np.ndarray,
+    mask: Optional[np.ndarray] = None,
 ) -> tuple[np.ndarray, np.ndarray]:
     """
     Compute scaled dot-product attention.
@@ -265,7 +268,11 @@ def feedforward_network(x: np.ndarray, hidden_dim: int = 3072) -> np.ndarray:
     hidden = x @ w1 + b1
 
     # GELU activation (approximation)
-    hidden = 0.5 * hidden * (1 + np.tanh(np.sqrt(2 / np.pi) * (hidden + 0.044715 * hidden**3)))
+    hidden = (
+        0.5
+        * hidden
+        * (1 + np.tanh(np.sqrt(2 / np.pi) * (hidden + 0.044715 * hidden**3)))
+    )
 
     # Second linear layer
     w2 = np.random.randn(hidden_dim, embedding_dim) * 0.02