git.diff

diff --git a/data/envs/metaworld/generate_dataset_all.sh b/data/envs/metaworld/generate_dataset_all.sh
index 802bf5c..7fe0781 100755
--- a/data/envs/metaworld/generate_dataset_all.sh
+++ b/data/envs/metaworld/generate_dataset_all.sh
@@ -1,29 +1,29 @@
 #!/bin/bash
 
 ENVS=(
-    assembly
-    basketball
-    bin-picking
-    box-close
-    button-press-topdown
-    button-press-topdown-wall
-    button-press
-    button-press-wall
-    coffee-button
-    coffee-pull
-    coffee-push
-    dial-turn
-    disassemble
-    door-close
-    door-lock
-    door-open
-    door-unlock
-    drawer-close
-    drawer-open
-    faucet-close
-    faucet-open
-    hammer
-    hand-insert
+    # assembly
+    # basketball
+    # bin-picking
+    # box-close
+    # button-press-topdown
+    # button-press-topdown-wall
+    # button-press
+    # button-press-wall
+    # coffee-button
+    # coffee-pull
+    # coffee-push
+    # dial-turn
+    # disassemble
+    # door-close
+    # door-lock
+    # door-open
+    # door-unlock
+    # drawer-close
+    # drawer-open
+    # faucet-close
+    # faucet-open
+    # hammer
+    # hand-insert
     handle-press-side
     handle-press
     handle-pull-side
diff --git a/data/envs/metaworld/push_all.sh b/data/envs/metaworld/push_all.sh
index 9d71467..26f8933 100755
--- a/data/envs/metaworld/push_all.sh
+++ b/data/envs/metaworld/push_all.sh
@@ -1,29 +1,29 @@
 #!/bin/bash
 
 ENVS=(
-    assembly
-    basketball
-    bin-picking
-    box-close
-    button-press-topdown
-    button-press-topdown-wall
-    button-press
-    button-press-wall
-    coffee-button
-    coffee-pull
-    coffee-push
-    dial-turn
-    disassemble
-    door-close
-    door-lock
-    door-open
-    door-unlock
-    drawer-close
-    drawer-open
-    faucet-close
-    faucet-open
-    hammer
-    hand-insert
+    # assembly
+    # basketball
+    # bin-picking
+    # box-close
+    # button-press-topdown
+    # button-press-topdown-wall
+    # button-press
+    # button-press-wall
+    # coffee-button
+    # coffee-pull
+    # coffee-push
+    # dial-turn
+    # disassemble
+    # door-close
+    # door-lock
+    # door-open
+    # door-unlock
+    # drawer-close
+    # drawer-open
+    # faucet-close
+    # faucet-open
+    # hammer
+    # hand-insert
     handle-press-side
     handle-press
     handle-pull-side
diff --git a/data/envs/metaworld/train_all.sh b/data/envs/metaworld/train_all.sh
index dbf328a..071e16a 100755
--- a/data/envs/metaworld/train_all.sh
+++ b/data/envs/metaworld/train_all.sh
@@ -1,29 +1,29 @@
 #!/bin/bash
 
 ENVS=(
-    assembly
-    basketball
-    bin-picking
-    box-close
-    button-press-topdown
-    button-press-topdown-wall
-    button-press
-    button-press-wall
-    coffee-button
-    coffee-pull
-    coffee-push
-    dial-turn
-    disassemble
-    door-close
-    door-lock
-    door-open
-    door-unlock
-    drawer-close
-    drawer-open
-    faucet-close
-    faucet-open
-    hammer
-    hand-insert
+    # assembly
+    # basketball
+    # bin-picking
+    # box-close
+    # button-press-topdown
+    # button-press-topdown-wall
+    # button-press
+    # button-press-wall
+    # coffee-button
+    # coffee-pull
+    # coffee-push
+    # dial-turn
+    # disassemble
+    # door-close
+    # door-lock
+    # door-open
+    # door-unlock
+    # drawer-close
+    # drawer-open
+    # faucet-close
+    # faucet-open
+    # hammer
+    # hand-insert
     handle-press-side
     handle-press
     handle-pull-side
@@ -55,4 +55,5 @@ ENVS=(
 
 for ENV in "${ENVS[@]}"; do
     python train.py --env $ENV-v2 --experiment $ENV-v2 --with_wandb True --wandb_user qgallouedec --wandb_project sample_facotry_metaworld
+    python enjoy.py --algo=APPO --env $ENV-v2 --experiment $ENV-v2 --train_dir=./train_dir --max_num_episodes=10 --push_to_hub --hf_repository=qgallouedec/sample-factory-$ENV-v2 --save_video --no_render --enjoy_script=enjoy --train_script=train --load_checkpoint_kind best
 done