Test gymnasium and Isaac Lab wrappers #184
Merged
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Toni-SM
merged commit 9e5b5e4
into
toni/fix_gymnasium_unwrapped_warning
0 of 2 checks passed
Toni-SM
added a commit
that referenced
this pull request
Aug 28, 2024
* Use the unwrapped environment to fix gymnasium warnings in PyTorch * Close the environment in the base class and add render return annotation * Update gymnasium wrapper * Fix gymnasium vectorized environments reset returns when called twice * Fix Isaac Lab environments reset returns when called twice * Fix Isaac Lab environments' state_space when num_states is zero * Test gymnasium and Isaac Lab wrappers (#184) * Add optional dependencies for tests * Add test for wrapping gymnasium environments in torch * Add edge case for vectorized environments * Add test for wrapping Isaac Lab environments in torch * Check for automatic wrapping and parametrize Isaac Lab wrapping test case * Use regular expression to detect environment types * Update gym wrapper * Add test for wrapping gym environments in torch * Apply pre-commit to tests folder * Fix auto wrapping for Isaac Gym preview 4 in torch * Update Isaac Gym preview wrapper * Add test for wrapping Isaac Gym preview environments in torch * Call base class's close method and rename internal variable * Update DeepMind wrapper * Add test for wrapping DeepMind environments in torch * Add test for wrapping Isaac Gym preview environments in torch * Update Omniverse Isaac Gym wrapper * Add test for wrapping Omniverse Isaac Gym environments in torch * Use regular expression to check for Omniverse Isaac Gym envs * Reflect wrapping and wrapper base class changes made in torch to jax * Update gymnasium wrapper in jax * Add test for wrapping gymnasium environments in jax * Update gym wrapper in jax * Add test for wrapping gym environments in jax * Apply pre-commits * Rename test files * Remove print statements * Update Isaac Gym preview wrapper in jax * Add test for wrapping Isaac Gym preview environments in jax * Apply pre-commit * Update Omniverse Isaac Gym wrapper in jax * Add test for wrapping Omniverse Isaac Gym environments in jax * Merge toni/multi-agent into toni/update_wrappers (#187) * Refactorize and simplify the MultiAgentEnvWrapper in PyTorch * Add Isaac Lab multi-agent wrappers in torch * Compute the multi-agend reward as the sum of each agent reward * Don't create memory tensors if memories parameter is None * Get environment state by calling the .state() method * Apply some changes from multi-agent branch to avoid merge conflict * Apply changes from torch implementation to jax implementation * Compute the multi-agend reward as the sum of each agent reward in jax * Fix current shared next states usage * Update Isaac Lab wrapper in jax * Update MultiAgentEnvWrapper in jax * Add test for wrapping Isaac Lab environments in jax * Add method to get the environment state in multi-agent base classes * Force base classes render and close methods to raise NotImplementedError * Update Isaac Lab loader args to the latest public version * Update the Petting Zoo wrapper in torch * Update Petting Zoo wrapper in jax * Add test for wrapping Petting Zoo environments * Update Bi-DexHands wrappers * Update Bi-DexHands MAPPO examples * Define environment wrapper device as a property * Update wrapping API in docs * Add Runner to utils table in docs * Add wrapper for Brax environments * Remove jax backend setup in runner * Add method to get environment state in Wrapper class * Use a default config for backward compatibility when runner cfg's memory field is not defined * Update environment wrapping image files * Update Isaac Gym and Omniverse Isaac Gym links * Update Isaac Gym and Omniverse Isaac Gym links in examples * Add wrapping snippets for Brax * Add Brax environment wrapping to docs * Add wrapping snippets for multi-agent envs * Update multi-agent wrapper in docs * Add test for brax environment wrapper * Handle numpy backend in brax envs wrapper in jax * Add Isaac Lab and Brax wrapper APIs to docs * Organize table in wrap_env docstrings * Update annotation in wrapper test files * Add Isaac Lab multi-agent wrapper test in torch * Fix Isaac Lab multi-agent environment in jax * Add Isaac Lab multi-agent wrapper test in jax * Remove IsaacLabSingleAgentWrapper class as the conversion feature is included in Isaac Lab * Update CHANGELOG
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
No description provided.