Code

Space-tensor conversion#206

Code

Merged

Toni-SM merged 63 commits intodevelopToni-SM/skrl:developfrom toni/space_tensor_conversionToni-SM/skrl:toni/space_tensor_conversionCopy head branch name to clipboardOct 18, 2024

+1,597-961

Commits

Commits on Sep 14, 2024

• Add utility to tensorize and flatten gymnasium spaces

  Toni-SM

  committed
  2645564

  Copy full SHA for 2645564

• Add spaces utility to docs

  Toni-SM

  committed
  7a1ea6b

  Copy full SHA for 7a1ea6b

Commits on Sep 18, 2024

• Add test for space utils implementations

  Toni-SM

  committed
  38ca1c8

  Copy full SHA for 38ca1c8

Commits on Sep 21, 2024

• Add funtion to sample spaces by batches

  Toni-SM

  committed
  c8fe715

  Copy full SHA for c8fe715

Commits on Sep 26, 2024

• Update test file

  Toni-SM

  committed
  65cbc82

  Copy full SHA for 65cbc82

Commits on Sep 29, 2024

• Allow None type

  Toni-SM

  committed
  3324048

  Copy full SHA for 3324048

• Update model defintions to support different input spaces

  Toni-SM

  committed
  45418b8

  Copy full SHA for 45418b8

• Update model definitions to support different input spaces

  Toni-SM

  committed
  85a1f0f

  Copy full SHA for 85a1f0f

• Merge branch 'toni/space_tensor_conversion' of github.com:Toni-SM/skrl into toni/space_tensor_conversion

  Toni-SM

  committed
  eff6e76

  Copy full SHA for eff6e76

• Update model definitions test file

  Toni-SM

  committed
  fe57d20

  Copy full SHA for fe57d20

Commits on Sep 30, 2024

• Replace view by reshape to support unclear operations

  Toni-SM

  committed
  f30e9f7

  Copy full SHA for f30e9f7

• Replace gym by gymnasium in wrapper base class annotation

  Toni-SM

  committed
  deb397c

  Copy full SHA for deb397c

• Use space utils to handle spaces in gymnasium-based env wrappers

  Toni-SM

  committed
  f3621c4

  Copy full SHA for f3621c4

• Replace gym with gymnasium in DeepMind wrapper test

  Toni-SM

  committed
  834e94a

  Copy full SHA for 834e94a

Commits on Oct 1, 2024

• Add utility to convert gym spaces to gymnasium spaces

  Toni-SM

  committed
  43bdc34

  Copy full SHA for 43bdc34

• Convert gym wrapper spaces to gymnasium

  Toni-SM

  committed
  a92ebed

  Copy full SHA for a92ebed

• Create a common test file for hypothesis' strategies

  Toni-SM

  committed
  2857927

  Copy full SHA for 2857927

• Add test for convert_gym_space space utils

  Toni-SM

  committed
  1f83016

  Copy full SHA for 1f83016

• Remove gymnasium.spaces.Sequence from supported spaces

  Toni-SM

  committed
  1579d1a

  Copy full SHA for 1579d1a

• Add space utility to untensorize spaces

  Toni-SM

  committed
  d96b13a

  Copy full SHA for d96b13a

Commits on Oct 2, 2024

• Update Isaac Gym preview wrapper to use space utils

  Toni-SM

  committed
  b241214

  Copy full SHA for b241214

• Update Omniverse Isaac Gym wrapper to use space utils

  Toni-SM

  committed
  fbde435

  Copy full SHA for fbde435

• Use spaces utils to process actions in Gym wrapper

  Toni-SM

  committed
  6a4b24d

  Copy full SHA for 6a4b24d

• Use spaces utils to process actions in Gymnasium wrapper

  Toni-SM

  committed
  ddbf5bb

  Copy full SHA for ddbf5bb

• Use spaces utils to process actions in Isaac Lab wrapper

  Toni-SM

  committed
  db9a114

  Copy full SHA for db9a114

• Raise exceptions when the spaces/values are not supported

  Toni-SM

  committed
  7e41f75

  Copy full SHA for 7e41f75

• Use spaces utils to process actions in Isaac Gym preview wrapper

  Toni-SM

  committed
  b5e27b2

  Copy full SHA for b5e27b2

• Use spaces utils to process actions in Omniverse Isaac Gym wrapper

  Toni-SM

  committed
  52ce622

  Copy full SHA for 52ce622

• Use spaces utils to process actions in PettingZoo wrapper

  Toni-SM

  committed
  80a8f02

  Copy full SHA for 80a8f02

• Use spaces utils to process actions in DeepMind wrapper

  Toni-SM

  committed
  5ab7777

  Copy full SHA for 5ab7777

• Allow to remove batch dimension when converting gym space

  Toni-SM

  committed
  abaa59e

  Copy full SHA for abaa59e

• Use spaces utils to process actions in Brax wrapper and allow operation with other spaces

  Toni-SM

  committed
  2e0182b

  Copy full SHA for 2e0182b

Commits on Oct 4, 2024

• Update spaces utils section in docs

  Toni-SM

  committed
  f77c9dc

  Copy full SHA for f77c9dc

Commits on Oct 5, 2024

• Replace each _get_space_size method call by the compute_space_size utility

  Toni-SM

  committed
  721efde

  Copy full SHA for 721efde

• Add static method to parse jax device

  Toni-SM

  committed
  ac92ba5

  Copy full SHA for ac92ba5

• Add space utility in jax

  Toni-SM

  committed
  05581c4

  Copy full SHA for 05581c4

• Add spaces utils test in jax

  Toni-SM

  committed
  856cc0d

  Copy full SHA for 856cc0d

• Rename spaces utility test in torch

  Toni-SM

  committed
  7ece2f6

  Copy full SHA for 7ece2f6

• Move spaces utils implementation in torch to its own file

  Toni-SM

  committed
  17c6f11

  Copy full SHA for 17c6f11

Commits on Oct 7, 2024

• Replace each _get_space_size method call by the compute_space_size utility in jax

  Toni-SM

  committed
  e003ada

  Copy full SHA for e003ada

• Add spaces utils API in jax to docs

  Toni-SM

  committed
  5f8b615

  Copy full SHA for 5f8b615

• Update model definitions to support different input spaces in jax

  Toni-SM

  committed
  722dfa2

  Copy full SHA for 722dfa2

• Update model instantiators definitions test file in jax

  Toni-SM

  committed
  a7d800f

  Copy full SHA for a7d800f

• Replace gym by gymnasium in wrapper base class annotation in jax

  Toni-SM

  committed
  07f3803

  Copy full SHA for 07f3803

Commits on Oct 8, 2024

• Add parameter to handle JAX and NumPy backends

  Toni-SM

  committed
  e5be9b8

  Copy full SHA for e5be9b8

• Update Gymnasium wrapper to use space utils in jax

  Toni-SM

  committed
  4fd8ffc

  Copy full SHA for 4fd8ffc

Commits on Oct 12, 2024

• Merge branch 'develop' into toni/space_tensor_conversion

  Toni-SM

  committed
  cf7e3bf

  Copy full SHA for cf7e3bf

• Merge branch 'toni/space_tensor_conversion' of github.com:Toni-SM/skrl into toni/space_tensor_conversion

  Toni-SM

  committed
  116140f

  Copy full SHA for 116140f

Commits on Oct 13, 2024

• Update Gym wrapper to use space utils in jax

  Toni-SM

  committed
  af41e69

  Copy full SHA for af41e69

• Update PettingZoo wrapper to use space utils in jax

  Toni-SM

  committed
  7364b86

  Copy full SHA for 7364b86

• Update Brax wrapper to use space utils in jax

  Toni-SM

  committed
  e59b7c5

  Copy full SHA for e59b7c5

• Update Isaac Lab wrapper to use space utils in jax

  Toni-SM

  committed
  80615af

  Copy full SHA for 80615af

• Update Omniverse Isaac Gym wrapper to use space utils in jax

  Toni-SM

  committed
  ead61e8

  Copy full SHA for ead61e8

• Update Isaac Gym preview wrapper to use space utils in jax

  Toni-SM

  committed
  155240b

  Copy full SHA for 155240b

• Move torch-based import statements

  Toni-SM

  committed
  56439e0

  Copy full SHA for 56439e0

• Tensorize and flatten Isaac Lab MARL env state

  Toni-SM

  committed
  582a668

  Copy full SHA for 582a668

• Add docstring to JAX config's parse_device function

  Toni-SM

  committed
  44ee0c1

  Copy full SHA for 44ee0c1

Commits on Oct 16, 2024

• Update docs and doctrings

  Toni-SM

  committed
  f7d5315

  Copy full SHA for f7d5315

• Cache env info in Omniverse Isaac Gym wrapper

  Toni-SM

  committed
  54545e6

  Copy full SHA for 54545e6

• Update CHANGELOG

  Toni-SM

  committed
  ffe2e53

  Copy full SHA for ffe2e53

• Improve CHANGELOG description

  Toni-SM

  committed
  39e6d28

  Copy full SHA for 39e6d28

• Improve JAX config parse_device implementation

  Toni-SM

  committed
  c41ad0f

  Copy full SHA for c41ad0f

Commits on Oct 18, 2024

• Use gymnsasium batch utility to sample fundamental spaces

  Toni-SM

  committed
  94083a9

  Copy full SHA for 94083a9