From a6f6b6b8966d810d476bfb61d5af73e66bfce0dd Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Sat, 1 Aug 2026 17:43:36 -0600 Subject: [PATCH] Fixed experiment opt name match --- toolkit/optimizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/toolkit/optimizer.py b/toolkit/optimizer.py index c6a3298..f1c86d8 100644 --- a/toolkit/optimizer.py +++ b/toolkit/optimizer.py @@ -103,7 +103,7 @@ def get_optimizer( elif lower_type == 'automagic3': from toolkit.optimizers.automagic3 import Automagic3 optimizer = Automagic3(params, lr=float(learning_rate), **optimizer_params) - elif lower_type == 'automagicEXPERIMENT': + elif lower_type == 'automagicexperiment': from toolkit.optimizers.automagicEXPERIMENT import AutomagicEXPERIMENT optimizer = AutomagicEXPERIMENT(params, lr=float(learning_rate), **optimizer_params) else: