From 05401e48dcd5bfbafb6f675896c6de8578d83780 Mon Sep 17 00:00:00 2001
From: Santhosh Janardhanan <santhoshj@gmail.com>
Date: Mon, 13 Apr 2026 17:16:16 -0400
Subject: [PATCH] fix: add --output-dir alias to forge train CLI

Added --output-dir as an alias for --output to match user expectations
and documentation. Fixes unrecognized arguments error.
---
 src/companion/forge/train.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/companion/forge/train.py b/src/companion/forge/train.py
index 86f604e..7682e19 100644
--- a/src/companion/forge/train.py
+++ b/src/companion/forge/train.py
@@ -211,7 +211,9 @@ def main():
         "--data", type=Path, required=True, help="Path to training data JSONL"
     )
     parser.add_argument(
+        "--output-dir",
         "--output",
+        dest="output",
         type=Path,
         default=Path("~/.companion/training"),
         help="Output directory",