From e8eb3bfdf36f765dc0631625608c3a4c58a29703 Mon Sep 17 00:00:00 2001
From: NanoCode012 <nano@axolotl.ai>
Date: Thu, 22 May 2025 18:14:44 +0700
Subject: [PATCH] fix: catch unhandled custom optimizer

---
 src/axolotl/core/trainer_builder/base.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/axolotl/core/trainer_builder/base.py b/src/axolotl/core/trainer_builder/base.py
index 6c3735256..90d6a2d9d 100644
--- a/src/axolotl/core/trainer_builder/base.py
+++ b/src/axolotl/core/trainer_builder/base.py
@@ -421,6 +421,10 @@ class TrainerBuilderBase(abc.ABC):
                 adam_kwargs["eps"] = (eps1, eps2)
 
                 optimizer_kwargs.update(adam_kwargs)
+            else:
+                raise ValueError(
+                    f"Unhandled optimizer: {self.cfg.optimizer}. Please raise an Issue."
+                )
 
             # Parse any additional optimizer args from config
             if self.cfg.optim_args: