From 4b8a8fae92b74aa090f3aeac5c4530756afd17ca Mon Sep 17 00:00:00 2001 From: Haroon Qureshi Date: Wed, 24 Jun 2026 07:44:03 -0700 Subject: [PATCH] Move code_builder.py to a common location. PiperOrigin-RevId: 937332995 Change-Id: I2f03581cc47d96337bb637f30c678060207333fd --- .../generators => python/mujoco/codegen}/code_builder.py | 0 wasm/codegen/generators/enums.py | 3 +-- wasm/codegen/generators/functions.py | 2 +- wasm/codegen/generators/structs.py | 2 +- wasm/codegen/tests/generators_test.py | 2 +- 5 files changed, 4 insertions(+), 5 deletions(-) rename {wasm/codegen/generators => python/mujoco/codegen}/code_builder.py (100%) diff --git a/wasm/codegen/generators/code_builder.py b/python/mujoco/codegen/code_builder.py similarity index 100% rename from wasm/codegen/generators/code_builder.py rename to python/mujoco/codegen/code_builder.py diff --git a/wasm/codegen/generators/enums.py b/wasm/codegen/generators/enums.py index e427b8b4..d1cf2abb 100644 --- a/wasm/codegen/generators/enums.py +++ b/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], diff --git a/wasm/codegen/generators/functions.py b/wasm/codegen/generators/functions.py index d8bb652c..72adf1f1 100644 --- a/wasm/codegen/generators/functions.py +++ b/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 diff --git a/wasm/codegen/generators/structs.py b/wasm/codegen/generators/structs.py index 197aac2d..8ec7d070 100644 --- a/wasm/codegen/generators/structs.py +++ b/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 diff --git a/wasm/codegen/tests/generators_test.py b/wasm/codegen/tests/generators_test.py index 3da3eb02..634473a9 100644 --- a/wasm/codegen/tests/generators_test.py +++ b/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