Skip to content

[WIP] Add support for online data streaming#158

Closed
Copilot wants to merge 1 commit into
mainfrom
copilot/add-support-for-online-data-streaming-again
Closed

[WIP] Add support for online data streaming#158
Copilot wants to merge 1 commit into
mainfrom
copilot/add-support-for-online-data-streaming-again

Conversation

Copy link
Copy Markdown
Contributor

Copilot AI commented Mar 4, 2026

  • Create embodichain/lab/engine/__init__.py and embodichain/lab/engine/data.py with OnlineDataEngine class
  • Add tensordict dependency to pyproject.toml
  • Update embodichain/lab/sim/types.py to use TensorDict types
  • Update embodichain/lab/sim/objects/articulation.py with active_joint_ids and active_dof
  • Update embodichain/lab/sim/objects/robot.py to return TensorDict from get_proprioception
  • Update embodichain/lab/sim/sensors/base_sensor.py to use TensorDict data buffer
  • Update embodichain/lab/sim/sensors/contact_sensor.py to use TensorDict data buffer
  • Update embodichain/lab/gym/envs/base_env.py with max_episode_steps, active_joint_ids, TensorDict support
  • Update embodichain/lab/gym/envs/embodied_env.py with rollout buffer, set_rollout_buffer, control_parts
  • Update embodichain/lab/gym/envs/managers/datasets.py with rollout buffer access
  • Update embodichain/lab/gym/envs/managers/observations.py with active_joint_ids
  • Update embodichain/lab/gym/envs/rl_env.py to remove deprecated episode_length
  • Update embodichain/lab/gym/utils/gym_utils.py with init_rollout_buffer_from_obs_action_space
  • Update embodichain/lab/gym/utils/registration.py to use base_env.max_episode_steps
  • Update embodichain/utils/configclass.py to remove episode_length
  • Update configs and docs

💡 You can make Copilot smarter by setting up custom instructions, customizing its development environment and configuring Model Context Protocol (MCP) servers. Learn more Copilot coding agent tips in the docs.

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