test: ./infra_tests.py train: ./infra_tests.py --train