nightly_build.yml

# This workflow will install Python dependencies, run tests and lint with a variety of Python versions
# For more information see: https://help.github.com/actions/language-and-framework-guides/using-python-with-github-actions

name: Push Binary Nightly

on:
  workflow_call:
    secrets:
#      AWS_PYTORCH_UPLOADER_ACCESS_KEY_ID:
#        required: true
#      AWS_PYTORCH_UPLOADER_SECRET_ACCESS_KEY:
#        required: true
      PYPI_TOKEN:
        required: false
  # run every day at 13:45am
  schedule:
    - cron:  '45 13 * * *'
  # or manually trigger it
  workflow_dispatch:
  push:
    branches:
      - "nightly"

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-wheel-linux:
    # Don't run on forked repos.
    if: github.repository_owner == 'pytorch'
    runs-on: ubuntu-20.04
    strategy:
      matrix:
        python_version: [
          ["3.9", "cp39-cp39"],
          ["3.10", "cp310-cp310"],
          ["3.11", "cp311-cp311"],
          ["3.12", "cp312-cp312"],
        ]
        cuda_support: [["", "cpu", "cpu"]]
    container: pytorch/conda-builder
    steps:
      - name: Checkout tensordict
        uses: actions/checkout@v4
      - name: Install PyTorch nightly
        run: |
          export PATH="/opt/python/${{ matrix.python_version[1] }}/bin:$PATH"
          python3 -c """import sys;print(sys.version)"""
          python3 -mpip install --pre torch --extra-index-url https://download.pytorch.org/whl/nightly/${{ matrix.cuda_support[1] }}
      - name: Build tensordict Nightly
        run: |
          rm -r dist || true
          export PATH="/opt/python/${{ matrix.python_version[1] }}/bin:$PATH"
          python3 -mpip install wheel
          python3 setup.py bdist_wheel \
            --package_name tensordict-nightly \
            --python-tag=${{ matrix.python-tag }}
           find dist -name '*whl' -exec bash -c ' mv $0 ${0/linux/manylinux1}' {} \;
      # pytorch/pytorch binaries are also manylinux_2_17 compliant but they
      # pretend that they're manylinux1 compliant so we do the same.
      - name: Show auditwheel output; confirm 2-17
        run: |
          python3 -mpip install auditwheel
          auditwheel show dist/*
      - name: Upload wheel for the test-wheel job
        uses: actions/upload-artifact@v4
        with:
          name: tensordict-linux-${{ matrix.python_version[0] }}_${{ matrix.cuda_support[2] }}.whl
          path: dist/*.whl

  upload-wheel-linux:
    # Don't run on forked repos.
    if: github.repository_owner == 'pytorch'
    needs: test-wheel-linux
    runs-on: ubuntu-20.04
    strategy:
      matrix:
        python_version: [
          ["3.9", "cp39-cp39"],
          ["3.10", "cp310-cp310"],
          ["3.11", "cp311-cp311"],
          ["3.12", "cp312-cp312"],
        ]
        cuda_support: [["", "cpu", "cpu"]]
    container: pytorch/conda-builder
    steps:
      - name: Checkout tensordict
        uses: actions/checkout@v4
      - name: Download built wheels
        uses: actions/download-artifact@v4
        with:
          name: tensordict-linux-${{ matrix.python_version[0] }}_${{ matrix.cuda_support[2] }}.whl
          path: /tmp/wheels
      - name: Push tensordict Binary to PYPI
        env:
            PYPI_TOKEN: ${{ secrets.PYPI_TOKEN }}
        run: |
          export PATH="/opt/python/${{ matrix.python_version[1] }}/bin:$PATH"
          python3 -mpip install twine
          python -m twine upload \
              --username __token__ \
              --password "$PYPI_TOKEN" \
              --skip-existing \
              /tmp/wheels/tensordict_nightly-*.whl \
              --verbose

  test-wheel-linux:
    # Don't run on forked repos.
    if: github.repository_owner == 'pytorch'
    needs: build-wheel-linux
    runs-on: ubuntu-20.04
    strategy:
      matrix:
        python_version: [
          ["3.9", "cp39-cp39"],
          ["3.10", "cp310-cp310"],
          ["3.11", "cp311-cp311"],
          ["3.12", "cp312-cp312"],
        ]
        cuda_support: [["", "cpu", "cpu"]]
    steps:
      - name: Setup Python
        uses: actions/setup-python@v4
        with:
          python-version: ${{ matrix.python_version[0] }}
          architecture: x64
        env:
          AGENT_TOOLSDIRECTORY: "/opt/hostedtoolcache"
      - name: Checkout tensordict
        uses: actions/checkout@v4
      - name: Install PyTorch Nightly
        run: |
          export PATH="/opt/python/${{ matrix.python_version[1] }}/bin:$PATH"
          python3 -mpip install --pre torch --extra-index-url https://download.pytorch.org/whl/nightly/${{ matrix.cuda_support[1] }}
      - name: Upgrade pip
        run: |
          export PATH="/opt/python/${{ matrix.python_version[1] }}/bin:$PATH"
          python3 -mpip install --upgrade pip
      - name: Install test dependencies
        run: |
          export PATH="/opt/python/${{ matrix.python_version[1] }}/bin:$PATH"
          python3 -mpip install numpy pytest pillow>=4.1.1 scipy networkx expecttest pyyaml
      - name: Download built wheels
        uses: actions/download-artifact@v4
        with:
          name: tensordict-linux-${{ matrix.python_version[0] }}_${{ matrix.cuda_support[2] }}.whl
          path: /tmp/wheels
        env:
          AGENT_TOOLSDIRECTORY: "/opt/hostedtoolcache"
      - name: Install built wheels
        run: |
          export PATH="/opt/python/${{ matrix.python_version[1] }}/bin:$PATH"
          python3 -mpip install /tmp/wheels/*
      - name: Log version string
        run: |
          # Avoid ambiguity of "import tensordict" by deleting the source files.
          rm -rf tensordict/
          export PATH="/opt/python/${{ matrix.python_version[1] }}/bin:$PATH"
          python3 -c "import tensordict; print(tensordict.__version__)"
      - name: Run tests
        run: |
          set -e
          export IN_CI=1
          mkdir test-reports
          export PATH="/opt/python/${{ matrix.python_version[1] }}/bin:$PATH"
          python3 -m torch.utils.collect_env
          python3 -c "import tensordict; print(tensordict.__version__);from tensordict.nn import TensorDictModule"
          EXIT_STATUS=0
          pytest test/smoke_test.py -v --durations 200
          exit $EXIT_STATUS

  build-wheel-windows:
    # Don't run on forked repos.
    if: github.repository_owner == 'pytorch'
    runs-on: windows-latest
    strategy:
      matrix:
        python_version: [
          ["3.9", "3.9"],
          ["3.10", "3.10.3"],
          ["3.11", "3.11"],
          ["3.12", "3.12"],
        ]
    steps:
      - name: Setup Python
        uses: actions/setup-python@v2
        with:
          python-version: ${{ matrix.python_version[1] }}
      - name: Checkout tensordict
        uses: actions/checkout@v4
      - name: Install PyTorch nightly
        shell: bash
        run: |
          python3 -mpip install --pre torch --extra-index-url https://download.pytorch.org/whl/nightly/cpu
      - name: Build tensordict nightly
        shell: bash
        run: |
          rm -r dist || true
          python3 -mpip install wheel
          python3 setup.py bdist_wheel \
            --package_name tensordict-nightly \
            --python-tag=${{ matrix.python-tag }}
      - name: Upload wheel for the test-wheel job
        uses: actions/upload-artifact@v4
        with:
          name: tensordict-win-${{ matrix.python_version[0] }}.whl
          path: dist/*.whl

  test-wheel-windows:
    # Don't run on forked repos.
    if: github.repository_owner == 'pytorch'
    needs: build-wheel-windows
    runs-on: windows-latest
    strategy:
      matrix:
        python_version: [
          ["3.9", "3.9"],
          ["3.10", "3.10.3"],
          ["3.11", "3.11"],
          ["3.12", "3.12"],
        ]
    steps:
      - name: Setup Python
        uses: actions/setup-python@v2
        with:
          python-version: ${{ matrix.python_version[1] }}
      - name: Checkout tensordict
        uses: actions/checkout@v4
      - name: Install PyTorch Nightly
        shell: bash
        run: |
          python3 -mpip install --pre torch --extra-index-url https://download.pytorch.org/whl/nightly/cpu
      - name: Upgrade pip
        shell: bash
        run: |
          python3 -mpip install --upgrade pip
      - name: Install test dependencies
        shell: bash
        run: |
          python3 -mpip install numpy pytest --no-cache-dir
      - name: Download built wheels
        uses: actions/download-artifact@v4
        with:
          name: tensordict-win-${{ matrix.python_version[0] }}.whl
          path: wheels
      - name: Install built wheels
        shell: bash
        run: |
          python3 -mpip install wheels/*
      - name: Log version string
        shell: bash
        run: |
          # Avoid ambiguity of "import tensordict" by deleting the source files.
          rm -rf tensordict/
          python3 -c "import tensordict; print(tensordict.__version__)"
      - name: Run tests
        shell: bash
        run: |
          set -e
          export IN_CI=1
          mkdir test-reports
          python -m torch.utils.collect_env
          python -c "import tensordict; print(tensordict.__version__);from tensordict.nn import TensorDictModule"
          EXIT_STATUS=0
          pytest test/smoke_test.py -v --durations 200
          exit $EXIT_STATUS

  upload-wheel-windows:
    # Don't run on forked repos.
    if: github.repository_owner == 'pytorch'
    needs: test-wheel-windows
    runs-on: windows-latest
    strategy:
      matrix:
        python_version: [
          ["3.9", "3.9"],
          ["3.10", "3.10.3"],
          ["3.11", "3.11"],
          ["3.12", "3.12"],
        ]
    steps:
      - name: Checkout tensordict
        uses: actions/checkout@v4
      - name: Download built wheels
        uses: actions/download-artifact@v4
        with:
          name: tensordict-win-${{ matrix.python_version[0] }}.whl
          path: wheels
      - name: Push tensordict Binary to PYPI
        shell: bash
        env:
            PYPI_TOKEN: ${{ secrets.PYPI_TOKEN }}
        run: |
          python3 -mpip install twine
          python3 -m twine upload \
              --username __token__ \
              --password "$PYPI_TOKEN" \
              --skip-existing \
              wheels/tensordict_nightly-*.whl \
              --verbose