Move code_builder.py to a common location.

PiperOrigin-RevId: 937332995
Change-Id: I2f03581cc47d96337bb637f30c678060207333fd

wasm/codegen/generators/enums.py

@@ -14,10 +14,9 @@
 
 """Generates Embind bindings for MuJoCo enums."""
 
+from mujoco.codegen import code_builder
 from introspect import ast_nodes
 
-from wasm.codegen.generators import code_builder
-
 
 def generate(
     enums: list[ast_nodes.EnumDecl],

wasm/codegen/generators/functions.py

@@ -16,9 +16,9 @@
 
 from typing import Tuple, cast
 
+from mujoco.codegen import code_builder
 from introspect import ast_nodes
 
-from wasm.codegen.generators import code_builder
 from wasm.codegen.generators import common
 from wasm.codegen.generators import constants
 

wasm/codegen/generators/structs.py

@@ -19,11 +19,11 @@ import dataclasses
 import math
 from typing import Tuple, Union, cast
 
+from mujoco.codegen import code_builder
 from introspect import ast_nodes
 from introspect import functions as introspect_functions
 from introspect import structs as introspect_structs
 
-from wasm.codegen.generators import code_builder
 from wasm.codegen.generators import common
 from wasm.codegen.generators import constants
 

wasm/codegen/tests/generators_test.py

@@ -15,9 +15,9 @@
 """Tests for helper modules."""
 
 from absl.testing import absltest
+from mujoco.codegen import code_builder
 from introspect import ast_nodes
 
-from wasm.codegen.generators import code_builder
 from wasm.codegen.generators import common
 from wasm.codegen.generators import enums
 from wasm.codegen.generators import functions