diff --git a/gen/drive3/src/api/mod.rs b/gen/drive3/src/api/mod.rs index a5b2a25052..aca6f89327 100644 --- a/gen/drive3/src/api/mod.rs +++ b/gen/drive3/src/api/mod.rs @@ -31,5 +31,5 @@ pub use method_builders::*; mod call_builders; pub use call_builders::*; -mod enums; -pub use enums::*; +pub mod enums; +pub(crate) use enums::*; diff --git a/src/generator/lib/enum_utils.py b/src/generator/lib/enum_utils.py index 98ddf573f5..c00d57d319 100644 --- a/src/generator/lib/enum_utils.py +++ b/src/generator/lib/enum_utils.py @@ -84,7 +84,7 @@ def find_enums_in_context(c: Context) -> list: for k, v in c.fqan_map.items(): # print(k) k = _parse_method_id(k, c) - if v.parameters: + if v.get('parameters'): for pk, pv in v.parameters.items(): enums = _add_enum_value(k, pk, pv, enums) return list(enums.values())