version: '1.0'

steps:

  BuildTorchANI:
    type: build
    description: Build TorchANI
    image_name: torchani
    dockerfile: Dockerfile
    tag: latest

  CodeStyle:
    image: '${{BuildTorchANI}}'
    commands:
      - flake8

  UnitTests:
    image: '${{BuildTorchANI}}'
    commands:
      - find . -name '*.pyc' -delete
      - python setup.py nosetests
      # - python2 setup.py test
      - codecov

  Tools:
    image: '${{BuildTorchANI}}'
    commands:
      - rm -rf *.pt
      - python tools/training-benchmark.py ./dataset/ani_gdb_s01.h5
      - python tools/neurochem-test.py ./dataset/ani_gdb_s01.h5
      - python tools/inference-benchmark.py --tqdm ./xyz_files/CH4-5.xyz

  ModuleMain:
    image: '${{BuildTorchANI}}'
    commands:
      - rm -rf *.pt
      - python -m torchani.neurochem.trainer --tqdm tests/test_data/inputtrain.ipt  dataset/ani_gdb_s01.h5 dataset/ani_gdb_s01.h5
      - python -m torchani.data.cache_aev tmp dataset/ani_gdb_s01.h5 256

  Docs:
    image: '${{BuildTorchANI}}'
    commands:
      - find . -name '*.pt' -delete
      - sphinx-build docs build