fix: add module type to loaded_modules when auto_register_xxx met…

…hod is called
ZihanType · Sep 5, 2023 · e88adbd · e88adbd
1 parent 0093e4a
commit e88adbd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/rudi/src/context.rs b/rudi/src/context.rs
@@ -1668,6 +1668,7 @@ impl ContextOptions {
 
         let mut cx = self.inner_create(|cx| {
             let module = ResolveModule::new::<AutoRegisterModule>();
+            cx.loaded_modules.push(module.ty());
             cx.load_providers(module.eager_create(), module.providers())
         });
 
@@ -1710,6 +1711,7 @@ impl ContextOptions {
 
         let mut cx = self.inner_create(|cx| {
             let module = ResolveModule::new::<AutoRegisterModule>();
+            cx.loaded_modules.push(module.ty());
             cx.load_providers(module.eager_create(), module.providers())
         });