From cbaee04dc17a2f272238ced4e93501421035ea1a Mon Sep 17 00:00:00 2001
From: n0k0m3 <10925298+n0k0m3@users.noreply.github.com>
Date: Fri, 10 Jan 2025 12:21:55 -0500
Subject: [PATCH] hydra compose register_amp_configs() fix
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Close #206

`register_amp_configs()` was called in main() register_decorator but missing in compose(). When user uses compose() to get config it will fail since it can’t find any registered amp defaults.
---
 modulus/sym/hydra/utils.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/modulus/sym/hydra/utils.py b/modulus/sym/hydra/utils.py
index 927c7ed8..6937815d 100644
--- a/modulus/sym/hydra/utils.py
+++ b/modulus/sym/hydra/utils.py
@@ -170,6 +170,7 @@ def compose(
     register_scheduler_configs()
     register_training_configs()
     register_modulus_configs()
+    register_amp_configs()
     register_graph_configs()
 
     cfg = hydra.compose(