Move timeout to colab

Author: maryamziaa

.github/workflows/colab.yml

@@ -16,7 +16,7 @@ on:
 
 jobs:
   colab:
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-22.04
     env:
         COLAB_ALWAYS_INSTALL_XVFB: 1
         QLEARNING_NUM_TRAINING_STEPS: 5
@@ -54,7 +54,7 @@ jobs:
         python -m pip install --progress-bar=off -e ./ml-agents
         python -m pip install --progress-bar=off -r colab_requirements.txt
     - name: Execute notebook
-      run: jupyter nbconvert --to notebook --execute --log-level=DEBUG --ExecutePreprocessor.kernel_name=python3 --ExecutePreprocessor.timeout=600 --output output-${{ matrix.notebook_path }} colab/${{ matrix.notebook_path }}
+      run: jupyter nbconvert --to notebook --execute --log-level=DEBUG --ExecutePreprocessor.kernel_name=python3 --output output-${{ matrix.notebook_path }} colab/${{ matrix.notebook_path }}
     - name: Upload colab results
       uses: actions/upload-artifact@v4
       with:

colab/Colab_UnityEnvironment_1_Run.ipynb

@@ -217,7 +217,7 @@
     "\n",
     "from mlagents_envs.registry import default_registry\n",
     "\n",
-    "env = default_registry[env_id].make()"
+    "env = default_registry[env_id].make(timeout_wait=120)"
    ],
    "execution_count": 3,
    "outputs": [