import subprocess
import sys
import os

cifar10_data_path = '/tmp/cifar10_data/'
cifar10_tarball_path = '/tmp/cifar10_data/cifar-10-binary.tar.gz'
if not os.path.exists(cifar10_tarball_path):
    subprocess.check_call(
        "aws s3 cp s3://zoox-testdata/tflight/cifar-10-binary.tar.gz %s && \
         tar xzf %s -C %s" % (cifar10_data_path, cifar10_tarball_path, cifar10_data_path),
        shell=True)

args = ["--model_name Piconet",
        "--model_path experimental/epronovost/meteor_offload/tflight_integration/piconet_tflight.py",
        "--input {}".format(cifar10_data_path)]

subprocess.check_call(
    "tflight/tfrun train {}".format(" ".join(args + sys.argv[1:])),
    shell=True)