Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Space-tensor conversion #206

Merged
merged 63 commits into from
Oct 18, 2024
Merged

Space-tensor conversion #206

merged 63 commits into from
Oct 18, 2024

Conversation

Toni-SM
Copy link
Owner

@Toni-SM Toni-SM commented Oct 5, 2024

No description provided.

@Toni-SM
Add utility to tensorize and flatten gymnasium spaces
2645564
@Toni-SM
Add spaces utility to docs
7a1ea6b
@Toni-SM
Add test for space utils implementations
38ca1c8
@Toni-SM
Add funtion to sample spaces by batches
c8fe715
@Toni-SM
Update test file
65cbc82
@Toni-SM
Allow None type
3324048
@Toni-SM
Update model defintions to support different input spaces
45418b8
@Toni-SM
Update model definitions to support different input spaces
85a1f0f
@Toni-SM
Merge branch 'toni/space_tensor_conversion' of github.com:Toni-SM/skr…
eff6e76 
…l into toni/space_tensor_conversion
@Toni-SM
Update model definitions test file
fe57d20
@Toni-SM
Replace view by reshape to support unclear operations
f30e9f7
@Toni-SM
Replace gym by gymnasium in wrapper base class annotation
deb397c
@Toni-SM
Use space utils to handle spaces in gymnasium-based env wrappers
f3621c4
@Toni-SM
Replace gym with gymnasium in DeepMind wrapper test
834e94a
@Toni-SM
Add utility to convert gym spaces to gymnasium spaces
43bdc34
@Toni-SM
Convert gym wrapper spaces to gymnasium
a92ebed
@Toni-SM
Create a common test file for hypothesis' strategies
2857927
@Toni-SM
Add test for convert_gym_space space utils
1f83016
@Toni-SM
Remove gymnasium.spaces.Sequence from supported spaces
1579d1a
@Toni-SM
Add space utility to untensorize spaces
d96b13a
@Toni-SM
Update Isaac Gym preview wrapper to use space utils
b241214
@Toni-SM
Update Omniverse Isaac Gym wrapper to use space utils
fbde435
@Toni-SM
Use spaces utils to process actions in Gym wrapper
6a4b24d
@Toni-SM
Use spaces utils to process actions in Gymnasium wrapper
ddbf5bb
@Toni-SM
Use spaces utils to process actions in Isaac Lab wrapper
db9a114
@Toni-SM
Raise exceptions when the spaces/values are not supported
7e41f75
@Toni-SM
Use spaces utils to process actions in Isaac Gym preview wrapper
b5e27b2
@Toni-SM
Use spaces utils to process actions in Omniverse Isaac Gym wrapper
52ce622
@Toni-SM
Use spaces utils to process actions in PettingZoo wrapper
80a8f02
@Toni-SM
Use spaces utils to process actions in DeepMind wrapper
5ab7777
Toni-SM added 28 commits October 5, 2024 17:41
@Toni-SM
Add space utility in jax
05581c4
@Toni-SM
Add spaces utils test in jax
856cc0d
@Toni-SM
Rename spaces utility test in torch
7ece2f6
@Toni-SM
Move spaces utils implementation in torch to its own file
17c6f11
@Toni-SM
Replace each _get_space_size method call by the compute_space_size ut…
e003ada 
…ility in jax
@Toni-SM
Add spaces utils API in jax to docs
5f8b615
@Toni-SM
Update model definitions to support different input spaces in jax
722dfa2
@Toni-SM
Update model instantiators definitions test file in jax
a7d800f
@Toni-SM
Replace gym by gymnasium in wrapper base class annotation in jax
07f3803
@Toni-SM
Add parameter to handle JAX and NumPy backends
e5be9b8
@Toni-SM
Update Gymnasium wrapper to use space utils in jax
4fd8ffc
@Toni-SM
Merge branch 'develop' into toni/space_tensor_conversion
cf7e3bf
@Toni-SM
Merge branch 'toni/space_tensor_conversion' of github.com:Toni-SM/skr…
116140f 
…l into toni/space_tensor_conversion
@Toni-SM
Update Gym wrapper to use space utils in jax
af41e69
@Toni-SM
Update PettingZoo wrapper to use space utils in jax
7364b86
@Toni-SM
Update Brax wrapper to use space utils in jax
e59b7c5
@Toni-SM
Update Isaac Lab wrapper to use space utils in jax
80615af
@Toni-SM
Update Omniverse Isaac Gym wrapper to use space utils in jax
ead61e8
@Toni-SM
Update Isaac Gym preview wrapper to use space utils in jax
155240b
@Toni-SM
Move torch-based import statements
56439e0
@Toni-SM
Tensorize and flatten Isaac Lab MARL env state
582a668
@Toni-SM
Add docstring to JAX config's parse_device function
44ee0c1
@Toni-SM
Update docs and doctrings
f7d5315
@Toni-SM
Cache env info in Omniverse Isaac Gym wrapper
54545e6
@Toni-SM
Update CHANGELOG
ffe2e53
@Toni-SM
Improve CHANGELOG description
39e6d28
@Toni-SM
Improve JAX config parse_device implementation
c41ad0f
@Toni-SM
Use gymnsasium batch utility to sample fundamental spaces
94083a9
@Toni-SM Toni-SM merged commit 934fbaa into develop Oct 18, 2024
2 checks passed
@Toni-SM Toni-SM deleted the toni/space_tensor_conversion branch October 18, 2024 20:42
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
1 participant
@Toni-SM