aboutsummaryrefslogtreecommitdiff
path: root/macros/src/codegen/module.rs
diff options
context:
space:
mode:
Diffstat (limited to 'macros/src/codegen/module.rs')
-rw-r--r--macros/src/codegen/module.rs4
1 files changed, 2 insertions, 2 deletions
diff --git a/macros/src/codegen/module.rs b/macros/src/codegen/module.rs
index 5f077a22..3634fcc0 100644
--- a/macros/src/codegen/module.rs
+++ b/macros/src/codegen/module.rs
@@ -111,7 +111,7 @@ pub fn codegen(ctxt: Context, resources_tick: bool, app: &App, extra: &Extra) ->
pub resources: Resources<#lt>
));
- let priority = if ctxt.is_init() {
+ let priority = if ctxt.is_init() || ctxt.is_generator(app) {
None
} else {
Some(quote!(priority))
@@ -281,7 +281,7 @@ pub fn codegen(ctxt: Context, resources_tick: bool, app: &App, extra: &Extra) ->
None
};
- let priority = if ctxt.is_init() {
+ let priority = if ctxt.is_init() || ctxt.is_generator(app) {
None
} else {
Some(quote!(priority: &#lt rtfm::export::Priority))