From 07f88f8057d1743f2a147468ad62c51bece783cd Mon Sep 17 00:00:00 2001
From: Jarrett <2613089+jjaareet@users.noreply.github.com>
Date: Thu, 16 Jan 2025 12:15:42 -0700
Subject: [PATCH] fix(lora): add back store_true default args (#1205)

---
 llms/mlx_lm/lora.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/llms/mlx_lm/lora.py b/llms/mlx_lm/lora.py
index 4d050bd5..43f508c3 100644
--- a/llms/mlx_lm/lora.py
+++ b/llms/mlx_lm/lora.py
@@ -78,6 +78,7 @@ def build_parser():
         "--train",
         action="store_true",
         help="Do training",
+        default=None,
     )
     parser.add_argument(
         "--data",
@@ -135,6 +136,7 @@ def build_parser():
         "--test",
         action="store_true",
         help="Evaluate on the test set after training",
+        default=None,
     )
     parser.add_argument(
         "--test-batches",
@@ -156,6 +158,7 @@ def build_parser():
         "--grad-checkpoint",
         action="store_true",
         help="Use gradient checkpointing to reduce memory use.",
+        default=None,
     )
     parser.add_argument("--seed", type=int, help="The PRNG seed")
     return parser