rl/.github/workflows/test-linux-mujoco.yml at main · pytorch/rl · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
name: MuJoCo Custom Envs Tests on Linux

on:
  pull_request:
    paths:
      - "torchrl/envs/custom/mujoco/**"
      - "test/libs/test_mujoco.py"
      - ".github/workflows/test-linux-mujoco.yml"
      - ".github/unittest/linux_libs/scripts_mujoco/**"
  push:
    branches:
      - nightly
      - main
      - release/*
    paths:
      - "torchrl/envs/custom/mujoco/**"
      - "test/libs/test_mujoco.py"
      - ".github/workflows/test-linux-mujoco.yml"
      - ".github/unittest/linux_libs/scripts_mujoco/**"
  workflow_dispatch:
  workflow_call:

concurrency:
  # Mirror the convention from test-linux-libs.yml: cancel in-progress runs
  # for the same ref, but let main runs all complete (so we can pinpoint
  # regressions to a specific commit if something breaks).
  group: test-linux-mujoco-${{ github.ref == 'refs/heads/main' && format('ci-master-{0}', github.sha) || format('ci-{0}', github.ref) }}
  cancel-in-progress: true

permissions:
  id-token: write
  contents: read

jobs:
  unittests-mujoco:
    strategy:
      matrix:
        python_version: ["3.11"]
        cuda_arch_version: ["12.8"]
    uses: pytorch/test-infra/.github/workflows/linux_job_v3.yml@main
    with:
      repository: pytorch/rl
      runner: "mt-l-x86aavx2-11-41-a10g"
      gpu-arch-type: cuda
      gpu-arch-version: "12.8"
      docker-image: "ghcr.io/pytorch/test-infra/osdc-cuda:cuda12.8.1-cudnn-devel-ubuntu24.04"
      timeout: 90
      script: |
        if [[ "${{ github.ref }}" =~ release/* ]]; then
          export RELEASE=1
          export TORCH_VERSION=stable
        else
          export RELEASE=0
          export TORCH_VERSION=nightly
        fi

        set -euo pipefail
        export PYTHON_VERSION="3.11"
        export CU_VERSION="cu128"
        export TAR_OPTIONS="--no-same-owner"
        export UPLOAD_CHANNEL="nightly"
        export TF_CPP_MIN_LOG_LEVEL=0
        export BATCHED_PIPE_TIMEOUT=60
        export TD_GET_DEFAULTS_TO_NONE=1

        nvidia-smi

        bash .github/unittest/linux_libs/scripts_mujoco/run_all.sh