SMDATAPARALLEL_PYTORCH_TEST_MNIST_SCRIPT = (
    "examples/pytorch/zero_code_change_examples/smdataparallel_mnist.py"
)
SMDATAPARALLEL_PYTORCH_TEST_MNIST_ARGS = ["--epochs", "1"]

SMDATAPARALLEL_TF2_TEST_MNIST_SCRIPT = "examples/tensorflow2/scripts/smdataparallel_mnist_tf2.py"

SMDATAPARALLEL_PYTORCH_TEST_MNIST_SCRIPT = (
    "examples/pytorch/zero_code_change_examples/smdataparallel_mnist.py"
)
SMDATAPARALLEL_PYTORCH_TEST_MNIST_ARGS = ["--epochs", "1"]

SMDATAPARALLEL_TF2_TEST_MNIST_SCRIPT = "examples/tensorflow2/scripts/smdataparallel_mnist_tf2.py"
