lightning/.run_local_tests.sh

15 lines
452 B
Bash

#!/usr/bin/env bash
# install APEX, see https://github.com/NVIDIA/apex#linux
# to imitate SLURM set only single node
export SLURM_LOCALID=0
# use this to run tests
rm -rf _ckpt_*
rm -rf ./lightning_logs
python -m coverage run --source pytorch_lightning -m py.test pytorch_lightning tests pl_examples -v --flake8
python -m coverage report -m
# specific file
# python -m coverage run --source pytorch_lightning -m py.test --flake8 --durations=0 -v -k