Code

feat: FlatPack environment#188

Code

Merged

clement-bonnet merged 91 commits intoinstadeepai:maininstadeepai/jumanji:mainfrom RuanJohn:implement-flatpack-environmentCopy head branch name to clipboardMar 13, 2024

+2,942-5

Commits

Commits on May 21, 2023

• feat: initial jigsaw commit.

  RuanJohn

  committed
  960ead0

  Copy full SHA for 960ead0

Commits on May 22, 2023

• feat: added puzzle numbers to env viewer

  RuanJohn

  committed
  0e6c994

  Copy full SHA for 0e6c994

• feat: initial code for random agent network.

  RuanJohn

  committed
  19f21f1

  Copy full SHA for 19f21f1

• feat: remove board action mask.

  RuanJohn

  committed
  3988f4d

  Copy full SHA for 3988f4d

• feat: add jigsaw random agent.

  RuanJohn

  committed
  14a53ed

  Copy full SHA for 14a53ed

• chore: change board_dim to num_rows and num_cols

  RuanJohn

  committed
  62625c0

  Copy full SHA for 62625c0

Commits on May 23, 2023

• feat: register environment and add random networks

  RuanJohn

  committed
  6828cfa

  Copy full SHA for 6828cfa

Commits on May 25, 2023

• feat: full action mask working.

  RuanJohn

  committed
  b81dda2

  Copy full SHA for b81dda2

• feat: cleaner action mask generation.

  RuanJohn

  committed
  f67315c

  Copy full SHA for f67315c

Commits on May 28, 2023

• feat: added jigsaw documentation

  RuanJohn

  committed
  941fee1

  Copy full SHA for 941fee1

• chore: typo fix.

  RuanJohn

  committed
  2839da0

  Copy full SHA for 2839da0

• feat: added class doctring to env.

  RuanJohn

  committed
  3474efa

  Copy full SHA for 3474efa

• feat: import jigsaw actor critic network.

  RuanJohn

  committed
  6378477

  Copy full SHA for 6378477

• wip: work on actor critic networks.

  RuanJohn

  committed
  478b504

  Copy full SHA for 478b504

• chore: better variable naming

  RuanJohn

  committed
  316e733

  Copy full SHA for 316e733

Commits on May 29, 2023

• chore: variable renaming in jigsaw networks.

  RuanJohn

  committed
  4e35b86

  Copy full SHA for 4e35b86

• chore: variable renaming in jigsaw networks.

  RuanJohn

  committed
  b681f83

  Copy full SHA for b681f83

• feat: jigsaw networks implemented.

  RuanJohn

  committed
  ee87ec3

  Copy full SHA for ee87ec3

• fix: fix action spec off by one.

  RuanJohn

  committed
  8416969

  Copy full SHA for 8416969

• feat: added jigsaw training config.

  RuanJohn

  committed
  5868835

  Copy full SHA for 5868835

• chore: minor fixes.

  RuanJohn

  committed
  4a8caad

  Copy full SHA for 4a8caad

• chore: fix action space in docs.

  Show description for d0aa02c
  

  RuanJohn

  and

  sash-a

  authored
  d0aa02c

  Copy full SHA for d0aa02c

• chore: docs action mask fix.

  Show description for e876908
  

  RuanJohn

  and

  sash-a

  authored
  e876908

  Copy full SHA for e876908

• chore: action mask fix in docs.

  Show description for 57b6a81
  

  RuanJohn

  and

  sash-a

  authored
  57b6a81

  Copy full SHA for 57b6a81

• chore: action mask fix in docs.

  Show description for 0267299
  

  RuanJohn

  and

  sash-a

  authored
  0267299

  Copy full SHA for 0267299

• chore: indent docstrings.

  RuanJohn

  committed
  6adfaf3

  Copy full SHA for 6adfaf3

• Merge branch 'main' into 143-implement-jigsaw-env

  RuanJohn

  authored
  6206819

  Copy full SHA for 6206819

Commits on May 30, 2023

• fix: action mask indexing bugfix.

  RuanJohn

  committed
  b344add

  Copy full SHA for b344add

• test: first flatpack experiments.

  RuanJohn

  committed
  416c137

  Copy full SHA for 416c137

Commits on May 31, 2023

• feat: rename jigsaw to flat_pack

  RuanJohn

  committed
  f5621bd

  Copy full SHA for f5621bd

• chore: more renaming

  RuanJohn

  committed
  66597cb

  Copy full SHA for 66597cb

Commits on Jun 4, 2023

• feat: rename types and remove unnecessary fields from the state.

  RuanJohn

  committed
  2161d2e

  Copy full SHA for 2161d2e

• chore: rename viewer.

  RuanJohn

  committed
  b1c8233

  Copy full SHA for b1c8233

• feat: reword environment to flatpack.

  RuanJohn

  committed
  0c30053

  Copy full SHA for 0c30053

• feat: add env training config.

  RuanJohn

  committed
  9692882

  Copy full SHA for 9692882

• feat: update docs.

  RuanJohn

  committed
  afad966

  Copy full SHA for afad966

• Merge main into flatpack

  RuanJohn

  committed
  95677f1

  Copy full SHA for 95677f1

• chore: method renaming.

  RuanJohn

  committed
  661368d

  Copy full SHA for 661368d

• chore: docstring indents.

  RuanJohn

  committed
  90f206c

  Copy full SHA for 90f206c

• chore: reward descriptions.

  RuanJohn

  committed
  bd39f9e

  Copy full SHA for bd39f9e

• chore: typo fix.

  RuanJohn

  committed
  cf30cb9

  Copy full SHA for cf30cb9

• feat: new env images.

  RuanJohn

  committed
  989dc58

  Copy full SHA for 989dc58

• feat: set default env.

  RuanJohn

  committed
  2f79724

  Copy full SHA for 2f79724

Commits on Jun 26, 2023

• Merge branch 'main' into flatpack-first-test

  RuanJohn

  committed
  580dfa5

  Copy full SHA for 580dfa5

• feat: added equally weighted block dense reward

  RuanJohn

  committed
  490974b

  Copy full SHA for 490974b

Commits on Jun 28, 2023

• feat: flatpack improved training networks.

  RuanJohn

  committed
  78a6bb9

  Copy full SHA for 78a6bb9

Commits on Jul 3, 2023

• Merge branch 'main' into flatpack-first-test

  RuanJohn

  committed
  3ad9f6a

  Copy full SHA for 3ad9f6a

Commits on Jul 14, 2023

• feat: correct final convolution and output projection in Unet.

  RuanJohn

  committed
  2aace92

  Copy full SHA for 2aace92

Commits on Jul 16, 2023

• chore: renamed variables

  RuanJohn

  committed
  6425261

  Copy full SHA for 6425261

• chore: minor test fixes.

  RuanJohn

  committed
  2d982ae

  Copy full SHA for 2d982ae

• fix: fix unet output bug.

  RuanJohn

  committed
  dc3baee

  Copy full SHA for dc3baee

Commits on Jul 17, 2023

• feat: added extra reward tests.

  RuanJohn

  committed
  29e3368

  Copy full SHA for 29e3368

• chore: removed todo comments.

  RuanJohn

  committed
  c88634a

  Copy full SHA for c88634a

• chore: cleaned up doctrings, variable names and comments in environment file.

  RuanJohn

  committed
  aa3c069

  Copy full SHA for aa3c069

• chore: fixed comments and docstrings in reward file.

  RuanJohn

  committed
  6f38714

  Copy full SHA for 6f38714

• chore: fixed comments and docstrings in reward file.

  RuanJohn

  committed
  d7cb940

  Copy full SHA for d7cb940

• chore: fixed comments and docstrings in utils file.

  RuanJohn

  committed
  ac7fae1

  Copy full SHA for ac7fae1

• feat: removed simplified env from registered environments.

  RuanJohn

  committed
  23c21b7

  Copy full SHA for 23c21b7

• chore: fixed typos in documentation.

  RuanJohn

  committed
  2a16dc7

  Copy full SHA for 2a16dc7

• chore: fixed flatpack actor-critic networks comments.

  RuanJohn

  committed
  e05dfea

  Copy full SHA for e05dfea

Commits on Jul 19, 2023

• feat: added new flatpack gif and png

  RuanJohn

  committed
  7e2805f

  Copy full SHA for 7e2805f

Commits on Aug 19, 2023

• Merge branch 'main' into implement-flatpack-environment

  clement-bonnet

  authored
  bade29b

  Copy full SHA for bade29b

Commits on Jan 10, 2024

• Merge branch 'main' into implement-flatpack-environment

  sash-a

  authored
  d638261

  Copy full SHA for d638261

Commits on Feb 14, 2024

• Merge branch 'main' into implement-flatpack-environment

  sash-a

  authored
  d1b5291

  Copy full SHA for d1b5291

Commits on Mar 7, 2024

• Merge branch 'main' into implement-flatpack-environment

  sash-a

  authored
  00e9f8f

  Copy full SHA for 00e9f8f

Commits on Mar 11, 2024

• chore: add action mask shape comment to types

  Show description for e83488d
  

  RuanJohn

  and

  sash-a

  authored
  e83488d

  Copy full SHA for e83488d

• chore: docstring formatting

  Show description for 747abfe
  

  RuanJohn

  and

  sash-a

  authored
  747abfe

  Copy full SHA for 747abfe

• chore: fix env name in config comments

  Show description for da20f74
  

  RuanJohn

  and

  sash-a

  authored
  da20f74

  Copy full SHA for da20f74

• chore: reset defualt training agent to random

  Show description for cb613e5
  

  RuanJohn

  and

  sash-a

  authored
  cb613e5

  Copy full SHA for cb613e5

• chore: docstring formatting fix

  Show description for aae8d1e
  

  RuanJohn

  and

  sash-a

  authored
  aae8d1e

  Copy full SHA for aae8d1e

• chore: docstring fix

  Show description for 6f1c9dd
  

  RuanJohn

  and

  sash-a

  authored
  6f1c9dd

  Copy full SHA for 6f1c9dd

• chore: grammar fix

  Show description for 67a77b6
  

  RuanJohn

  and

  sash-a

  authored
  67a77b6

  Copy full SHA for 67a77b6

• chore: docstring fomatting

  Show description for 2ab98ed
  

  RuanJohn

  and

  sash-a

  authored
  2ab98ed

  Copy full SHA for 2ab98ed

• chore: docstring formatting fix

  Show description for bdb11f3
  

  RuanJohn

  and

  sash-a

  authored
  bdb11f3

  Copy full SHA for bdb11f3

• chore: remove manually setting vmap in_axes to defaults

  Show description for 1055147
  

  RuanJohn

  and

  sash-a

  authored
  1055147

  Copy full SHA for 1055147

• chore: remove manually setting vmap in_axes to defaults

  Show description for 627ca7c
  

  RuanJohn

  and

  sash-a

  authored
  627ca7c

  Copy full SHA for 627ca7c

• chore: code formatting

  Show description for e9e1a97
  

  RuanJohn

  and

  sash-a

  authored
  e9e1a97

  Copy full SHA for e9e1a97

• Merge branch 'main' into implement-flatpack-environment

  RuanJohn

  authored
  9d463fc

  Copy full SHA for 9d463fc

• chore: rename current_grid to grid and make grid the first attribute in timestep and state

  RuanJohn

  committed
  a05b35f

  Copy full SHA for a05b35f

• chore: remove jnp.where from _get_ones_like_expanded_block method

  RuanJohn

  committed
  c08fc62

  Copy full SHA for c08fc62

• chore: remove __init__ from random flatpack generator

  RuanJohn

  committed
  6a8a77b

  Copy full SHA for 6a8a77b

• chore: rename nibs -> interlock

  RuanJohn

  committed
  ae48129

  Copy full SHA for ae48129

• chore: add comment to explain why grid_with_block is initially empty

  RuanJohn

  committed
  8ff3ee1

  Copy full SHA for 8ff3ee1

• Merge branch 'main' into implement-flatpack-environment

  RuanJohn

  authored
  dd8cb49

  Copy full SHA for dd8cb49

Commits on Mar 12, 2024

• chore: remove sparse reward

  RuanJohn

  committed
  dbc8a8d

  Copy full SHA for dbc8a8d

• chore: rename env test methods

  RuanJohn

  committed
  69ce651

  Copy full SHA for 69ce651

• feat: add flat_pack documentation links to mkdocs.yml

  RuanJohn

  committed
  fc0d1c0

  Copy full SHA for fc0d1c0

• chore rename action to placed_block in reward classes

  RuanJohn

  committed
  6d4a797

  Copy full SHA for 6d4a797

• feat: determine if action is legal from the state action mask

  RuanJohn

  committed
  9e2c6b0

  Copy full SHA for 9e2c6b0

Commits on Mar 13, 2024

• chore: suggestions from review

  Show description for 95389a6
  

  3 peopleauthored
  95389a6

  Copy full SHA for 95389a6

• chore: change obs and state type to int

  sash-a

  committed
  a87bfd7

  Copy full SHA for a87bfd7