Fixed failing GPU test. #488
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: colab | |
| on: | |
| pull_request: | |
| paths: # This action will only run if the PR modifies a file in one of these directories | |
| - 'ml-agents-envs/**' | |
| - 'gym-unity/**' | |
| - 'colab/**' | |
| - '.github/workflows/colab.yml' | |
| push: | |
| branches: | |
| - main | |
| - develop | |
| - 'release/**' | |
| workflow_dispatch: | |
| jobs: | |
| colab: | |
| runs-on: ubuntu-latest | |
| env: | |
| COLAB_ALWAYS_INSTALL_XVFB: 1 | |
| QLEARNING_NUM_TRAINING_STEPS: 5 | |
| QLEARNING_NUM_NEW_EXP: 64 | |
| QLEARNING_BUFFER_SIZE: 64 | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| notebook_path: [Colab_UnityEnvironment_1_Run.ipynb, Colab_UnityEnvironment_2_Train.ipynb, Colab_UnityEnvironment_3_SideChannel.ipynb] | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - name: Set up Python | |
| uses: actions/setup-python@v5 | |
| with: | |
| python-version: 3.10.12 | |
| - uses: actions/setup-dotnet@v4 | |
| with: | |
| dotnet-version: '6.0.x' | |
| - name: Cache pip | |
| uses: actions/cache@v4 | |
| with: | |
| # This path is specific to Ubuntu | |
| path: ~/.cache/pip | |
| # Look to see if there is a cache hit for the corresponding requirements file | |
| key: ${{ runner.os }}-pip-${{ hashFiles('ml-agents/setup.py', 'ml-agents-envs/setup.py', 'gym-unity/setup.py', 'colab_requirements.txt') }} | |
| restore-keys: | | |
| ${{ runner.os }}-pip- | |
| ${{ runner.os }}- | |
| - name: Install dependencies | |
| run: | | |
| python -m pip install --upgrade pip | |
| python -m pip install --upgrade setuptools | |
| # Install the local checkouts of ml-agents. This will prevent the colab notebooks from installing a released version. | |
| python -m pip install --progress-bar=off -e ./ml-agents-envs | |
| 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 --output output-${{ matrix.notebook_path }} colab/${{ matrix.notebook_path }} | |
| - name: Upload colab results | |
| uses: actions/upload-artifact@v4 | |
| with: | |
| name: artifacts-${{ matrix.notebook_path }} | |
| path: | | |
| colab/output-${{ matrix.notebook_path }} | |
| # Use always() to always run this step to publish execution results when there are failures | |
| if: ${{ always() }} |