## BabyAI Instructions

1. **Download the dataset**  
   Run the following command to download the Minari dataset:
   ```bash
   minari download minigrid/BabyAI-GoTo/optimal-fullobs-v0
    ```
2. **Pretrain on the dataset**
    Run the following command:
    ```bash
    python pretrain.py --game_code goto 
    ```

3. **Inspect the dataset**
    Use the provided script to inspect dataset details (including environment seeds):
    ```bash
    python inspect_minari.py --game_code goto --dataset_dir file
    ```

4. **Print seeds**
    To print the train or evaluate seeds in .txt file, run:
    ```bash
    python print_seeds.py seeds/csv_file
    ```
5. **Update config files**
    Insert the extracted train and evaluate seeds into the corresponding config files.

6. **Finetune the pretrained policy**
    For example, to finetune on polychromic ppo, run:
    ```bash
    python poly_ppo.py --game_code goto
    ```

7. **Evaluate the policy using pass@n**
    ```bash
    python evaluate_pass_n.py --game_code goto --weights policy_path
    ```

8. **Test Generalization**
    ```bash
    python test_generalization.py --game_code goto --pretrained_weights pretrain.pt --finetuned_weights finetune.pt
    ```