fix code format

Signed-off-by: Prathamesh Mandke <46148373+pkmandke@users.noreply.github.com>

Author: pkmandke

python/src/coreai_models/llm/export.py

@@ -325,8 +325,7 @@ def _resolve_export_config(args: argparse.Namespace) -> ExportConfig:
 
     if args.disable_embedding_quantization_ios and variant != "iOS":
         raise SystemExit(
-            "--disable-embedding-quantization-ios requires --platform iOS "
-            f"(got '{variant}')."
+            f"--disable-embedding-quantization-ios requires --platform iOS (got '{variant}')."
         )
 
     if args.compression_config is not None:
@@ -430,9 +429,7 @@ def main() -> None:
             print(f"  num_layers:         {config.num_layers}")
         print(f"  overwrite:          {config.overwrite}")
         if config.variant == "iOS":
-            print(
-                f"  disable_embedding_quantization: {config.disable_embedding_quantization}"
-            )
+            print(f"  disable_embedding_quantization: {config.disable_embedding_quantization}")
         return
 
     result = export_model(config)

python/src/coreai_models/primitives/ios/sdpa.py

@@ -78,18 +78,8 @@ def forward(
             )
 
             q = query.reshape(B, n_heads, self.head_dim, S).transpose(2, 3).contiguous()
-            k = (
-                key[..., :S]
-                .reshape(B, n_kv_heads, self.head_dim, S)
-                .transpose(2, 3)
-                .contiguous()
-            )
-            v = (
-                value[..., :S]
-                .reshape(B, n_kv_heads, self.head_dim, S)
-                .transpose(2, 3)
-                .contiguous()
-            )
+            k = key[..., :S].reshape(B, n_kv_heads, self.head_dim, S).transpose(2, 3).contiguous()
+            v = value[..., :S].reshape(B, n_kv_heads, self.head_dim, S).transpose(2, 3).contiguous()
 
             out = F.scaled_dot_product_attention(
                 q,