tdpeuter/2026SEL3-project-Brittle_Star_Controller_Modularity
tdpeuter/2026SEL3-project-Brittle_Star_Controller_Modularity
Archived
1
Fork 0
Code Issues Releases Activity

feat: tool to construct XML

Browse source
This commit is contained in:
Tibo De Peuter 2026-04-28 16:42:46 +02:00
parent 5716b73eb9
commit 33fbc4c96f
Signed by: tdpeuter
SSH key fingerprint: SHA256:u/h/LVoqKF1Iz02uOyxe6hcjmoZASCGV2HM0TG9ZMoU
2 changed files with 188 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/test_morphology_render.py
View file

@ -14,7 +14,7 @@ from brittle_star_project.environment.BrittleStarJaxEnvWrapper import BrittleSta
@pytest.mark.skipif(os.getenv("CI") == "true", reason="No OpenGL display in CI")
def test_render_morphologies():
base_dir = "runs/renders"
base_dir = "runs/morphologies"
os.makedirs(base_dir, exist_ok=True)
# --- 1. Full 5-Arm Morphology ---
@ -34,7 +34,7 @@ def test_render_morphologies():
renderer_full = mujoco.Renderer(model=model_full)
renderer_full.update_scene(data_full, camera=1)
pixels_full = renderer_full.render()
image_path = os.path.join(base_dir, "full_5_arm.png")
image_path = os.path.join(base_dir, "5_arm.png")
Image.fromarray(pixels_full).save(image_path)
print(f"Generated full morphology render: {image_path}")