pytest_gpu:
  name: Pytest GPU
  agent:
    type: Unity::VM::GPU
    image: package-ci/ubuntu:v1.3.1-719011
    flavor: b1.large
  commands:
    - |
      sudo apt-get update && sudo apt-get install -y python3-venv
      python3 -m venv venv && source venv/bin/activate
      python3 -m pip install pyyaml --index-url https://artifactory.prd.it.unity3d.com/artifactory/api/pypi/pypi/simple
      python3 -u -m ml-agents.tests.yamato.setup_venv
      python3 -m pip install --progress-bar=off -r test_requirements.txt --index-url https://artifactory.prd.it.unity3d.com/artifactory/api/pypi/pypi/simple
      python3 -m pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html --index-url https://artifactory.prd.it.unity3d.com/artifactory/api/pypi/pypi/simple
      python3 -m pytest -m "not check_environment_trains" --junitxml=junit/test-results.xml -p no:warnings
  triggers:
    cancel_old_ci: true
    expression: |
      (push.branch eq "main" OR
      push.branch match "release.+") AND
      push.changes.any match "ml-agents/**" AND
      NOT push.changes.all match "**/*.md"
  artifacts:
    logs:
      paths:
        - "artifacts/standalone_build.txt"