# https://docs.drone.io/pipeline/docker/examples/languages/python/#python-example

kind: pipeline
type: docker
name: torch-GPU

steps:
- name: testing
  image: pytorch/pytorch:1.4-cuda10.1-cudnn7-runtime

  environment:
    SLURM_LOCALID: 0
    CODECOV_TOKEN:
      from_secret: codecov_token

  #volumes:
  #  # Mount pip cache from host
  #  - name: pip_cache
  #    path: /opt/conda/lib/python3.7/site-packages

  commands:
    - python --version
    - pip install pip -U
    - pip --version
    - nvidia-smi
    - bash ./tests/install_AMP.sh
    - pip install -r requirements.txt --user -q
    - pip install coverage pytest pytest-cov pytest-flake8 codecov -q
    - pip install -r ./tests/requirements.txt --user -q
    - pip list
    - python -c "import torch ; print(' & '.join([torch.cuda.get_device_name(i) for i in range(torch.cuda.device_count())]) if torch.cuda.is_available() else 'only CPU')"
    - coverage run --source pytorch_lightning -m py.test pytorch_lightning tests benchmarks -v --doctest-modules #  --flake8
    - coverage report
    - codecov --token $CODECOV_TOKEN  # --pr $DRONE_PULL_REQUEST --build $DRONE_BUILD_NUMBER --branch $DRONE_BRANCH --commit $DRONE_COMMIT --tag $DRONE_TAG
    - python tests/collect_env_details.py

trigger:
  branch:
    - master
  event:
    include:
    - push
    - pull_request

#volumes:
#  - name: pip_cache
#    host:
#      path: /tmp/cache/drone/pip