[Algorithm] Added TQC #6577
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
| # This workflow builds the torchrl docs and deploys them to gh-pages. | |
| name: Generate documentation | |
| on: | |
| push: | |
| branches: | |
| - main | |
| pull_request: | |
| branches: | |
| - "*" | |
| workflow_dispatch: | |
| concurrency: | |
| # Documentation suggests ${{ github.head_ref }}, but that's only available on pull_request/pull_request_target triggers, so using ${{ github.ref }}. | |
| # On master, we want all builds to complete even if merging happens faster to make it easier to discover at which point something broke. | |
| group: ${{ github.workflow }}-${{ github.ref == 'refs/heads/main' && format('ci-master-{0}', github.sha) || format('ci-{0}', github.ref) }} | |
| cancel-in-progress: true | |
| jobs: | |
| build_docs_job: | |
| runs-on: linux.g5.4xlarge.nvidia.gpu | |
| defaults: | |
| run: | |
| shell: bash -l {0} | |
| container: nvidia/cudagl:11.4.0-runtime | |
| steps: | |
| - name: Install deps | |
| run: | | |
| apt-get update && apt-get install -y git wget gcc g++ | |
| - name: Checkout | |
| uses: actions/checkout@v3 | |
| # Update references | |
| - name: Setup conda | |
| run: | | |
| root_dir="$(pwd)" | |
| conda_dir="${root_dir}/conda" | |
| env_dir="${root_dir}/env" | |
| os=Linux | |
| # 1. Install conda at ./conda | |
| printf "* Installing conda\n" | |
| wget -O miniconda.sh "http://repo.continuum.io/miniconda/Miniconda3-latest-${os}-x86_64.sh" | |
| bash ./miniconda.sh -b -f -p "${conda_dir}" | |
| eval "$(${conda_dir}/bin/conda shell.bash hook)" | |
| printf "* Creating a test environment\n" | |
| conda create --prefix "${env_dir}" -y python=3.8 | |
| printf "* Activating\n" | |
| conda activate "${env_dir}" | |
| - name: Update pip | |
| run: | | |
| apt-get install python3.8 python3-pip -y | |
| pip3 install --upgrade pip | |
| - name: check python version | |
| run: | | |
| python3 --version | |
| - name: Check git version | |
| run: git version | |
| - name: Install PyTorch | |
| shell: bash | |
| run: | | |
| pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu --quiet --root-user-action=ignore | |
| #pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu118 --quiet --root-user-action=ignore | |
| - name: Install tensordict | |
| run: | | |
| pip3 install git+https://github.com/pytorch/tensordict.git --quiet --root-user-action=ignore | |
| - name: Install TorchRL | |
| run: | | |
| python3 setup.py develop | |
| - name: Install requirements | |
| run: | | |
| pip3 install -r docs/requirements.txt --quiet --root-user-action=ignore | |
| - name: Test torchrl installation | |
| shell: bash | |
| run: | | |
| mkdir _tmp | |
| cd _tmp | |
| PYOPENGL_PLATFORM=egl MUJOCO_GL=egl python3 -c """from torchrl.envs.libs.dm_control import DMControlEnv | |
| print(DMControlEnv('cheetah', 'run', from_pixels=True).reset())""" | |
| cd .. | |
| - name: Build the docset | |
| id: build_doc | |
| run: | | |
| cd ./docs | |
| # timeout 7m bash -ic "MUJOCO_GL=egl sphinx-build ./source _local_build" || code=$?; if [[ $code -ne 124 && $code -ne 0 ]]; then exit $code; fi | |
| bash -ic "PYOPENGL_PLATFORM=egl MUJOCO_GL=egl sphinx-build ./source _local_build" || code=$?; if [[ $code -ne 124 && $code -ne 0 ]]; then exit $code; fi | |
| # PYOPENGL_PLATFORM=egl MUJOCO_GL=egl sphinx-build ./source _local_build | |
| cd .. | |
| - name: Install rsync 📚 | |
| run: | | |
| apt-get update && apt-get install -y rsync | |
| - name: Pull TensorDict docs | |
| run: | | |
| git clone --branch gh-pages https://github.com/pytorch/tensordict.git docs/_local_build/tensordict | |
| rm -rf docs/_local_build/tensordict/.git | |
| - name: Get output time | |
| run: echo "The time was ${{ steps.build.outputs.time }}" | |
| - name: Upload wheel for download | |
| uses: actions/upload-artifact@v2 | |
| with: | |
| name: build | |
| path: docs/_local_build/ | |
| - name: Deploy | |
| if: ${{ github.ref == 'refs/heads/main' || github.event_name == 'workflow_dispatch' }} | |
| uses: JamesIves/github-pages-deploy-action@releases/v4 | |
| with: | |
| token: ${{ secrets.GITHUB_TOKEN }} | |
| branch: gh-pages # The branch the action should deploy to. | |
| folder: docs/_local_build/ # The folder the action should deploy. | |
| CLEAN: false |