Apply black formatting

Author: codinglabsong

scripts/sample_data.py

@@ -11,10 +11,10 @@ def parse_args() -> argparse.Namespace:
     Parse command-line arguments.
 
     Returns:
-        argparse.Namespace:  
-            - input (str): Path to the full CSV file to sample from.  
-            - output (str): Path where the sampled CSV will be written.  
-            - n (int): Number of examples to sample.  
+        argparse.Namespace:
+            - input (str): Path to the full CSV file to sample from.
+            - output (str): Path where the sampled CSV will be written.
+            - n (int): Number of examples to sample.
             - seed (int): Random seed for reproducibility.
     """
     p = argparse.ArgumentParser(
@@ -41,11 +41,7 @@ def parse_args() -> argparse.Namespace:
     return p.parse_args()
 
 
-def sample_dataset(
-    input_csv: Path, 
-    output_csv: Path,
-    sample_size: int, 
-    seed: int = 42):
+def sample_dataset(input_csv: Path, output_csv: Path, sample_size: int, seed: int = 42):
     """
     Load a CSV, draw a random sample, and write it out.
 

src/bart_reddit_lora/model.py

@@ -28,7 +28,14 @@ def build_peft_model(
     r: int = 8,
     lora_dropout: float = 0.1,
     bias: str = "none",
-    target_modules: Sequence[str] = ("q_proj", "k_proj", "v_proj", "o_proj", "fc1", "fc2"),
+    target_modules: Sequence[str] = (
+        "q_proj",
+        "k_proj",
+        "v_proj",
+        "o_proj",
+        "fc1",
+        "fc2",
+    ),
     modules_to_save: Sequence[str] = ("lm_head",),
 ) -> PeftModel:
     """Wrap a base BART model with a LoRA adapter configuration.