Skip to content

Add cpp rollout and sim#88

Draft
jbruedigam-bdai wants to merge 1 commit intojbruedigam/generic_task_spacefrom
jbruedigam/specialized_sim_rollout
Draft

Add cpp rollout and sim#88
jbruedigam-bdai wants to merge 1 commit intojbruedigam/generic_task_spacefrom
jbruedigam/specialized_sim_rollout

Conversation

@jbruedigam-bdai
Copy link
Copy Markdown
Collaborator

No description provided.

Copy link
Copy Markdown
Collaborator

@alberthli alberthli left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Added a couple of comments on quick skim. If it's determined that this will be merged, I probably shouldn't be the one who approves, since a lot of the cpp code originated from me.

Comment thread judo/tasks/cylinder_push_new.py Outdated
Comment on lines +46 to +47
self.RolloutBackend = RolloutBackend
self.SimBackend = SimBackend
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

isn't this in the base task class? also, this file should be removed anyway before merging into main

Comment thread judo/utils/mujoco_cpp.py
Comment on lines +14 to +20
def __init__(self, num_threads: int, backend: Literal["mujoco"], task_to_sim_ctrl: Callable) -> None:
"""Initialize the backend with a number of threads."""
self.backend = backend
if self.backend == "mujoco":
self.setup_mujoco_backend()
else:
raise ValueError(f"Unknown backend: {self.backend}")
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

we should probably combine this with the existing RolloutBackend infra and just add a new element to the Literal type that's like "mujoco_custom" or something

@jbruedigam-bdai jbruedigam-bdai force-pushed the jbruedigam/specialized_sim_rollout branch from 5e5cabe to e945d19 Compare September 10, 2025 11:51
@jbruedigam-bdai jbruedigam-bdai force-pushed the jbruedigam/specialized_sim_rollout branch from e945d19 to 94d4399 Compare September 10, 2025 12:14
johnzhang-rai pushed a commit to johnzhang3/judo that referenced this pull request Oct 21, 2025
…project.toml and pixi.lock by taking upstream versions
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants