def test_invalid_tpu_chief_config(self):
with pytest.raises(ValueError, match=r"Invalid `chief_config`"):
validate.validate(
entry_point="tests/testdata/mnist_example_using_fit.py",
distribution_strategy="auto",
requirements_txt="tests/testdata/requirements.txt",
chief_config=machine_config.COMMON_MACHINE_CONFIGS["TPU"],
worker_config=machine_config.COMMON_MACHINE_CONFIGS["K80_1X"],
worker_count=1,
region="us-central1",
args=None,
stream_logs=True,
docker_image_bucket_name=None,
called_from_notebook=False,
)
def test_invalid_cloud_bucket_name(self):
with pytest.raises(ValueError, match=r"Invalid `docker_image_bucket_name`"):
validate.validate(
entry_point="tests/testdata/mnist_example_using_fit.py",
distribution_strategy="auto",
requirements_txt="tests/testdata/requirements.txt",
chief_config=TestValidate.VALID_MACHINE_CONFIG,
worker_config=TestValidate.VALID_MACHINE_CONFIG,
worker_count=1,
region="us-central1",
args=None,
stream_logs=False,
docker_image_bucket_name=None,
called_from_notebook=True,
)
def test_valid_args(self):
validate.validate(
entry_point="tests/testdata/mnist_example_using_fit.py",
distribution_strategy="auto",
requirements_txt="tests/testdata/requirements.txt",
chief_config=machine_config.COMMON_MACHINE_CONFIGS["K80_1X"],
worker_config=machine_config.COMMON_MACHINE_CONFIGS["K80_1X"],
worker_count=1,
region="us-central1",
args=None,
stream_logs=True,
docker_image_bucket_name=None,
called_from_notebook=False,
)
validate.validate(
entry_point="tests/testdata/mnist_example_using_fit.py",
distribution_strategy=None,
requirements_txt="tests/testdata/requirements.txt",
chief_config=machine_config.COMMON_MACHINE_CONFIGS["K80_1X"],
worker_config=None,
worker_count=0,
region="us-central1",
args=["1000"],
stream_logs=False,
docker_image_bucket_name=None,
called_from_notebook=False,
)
validate.validate(
entry_point="tests/testdata/mnist_example_using_fit.ipynb",
distribution_strategy=None,
requirements_txt="tests/testdata/requirements.txt",
chief_config=machine_config.COMMON_MACHINE_CONFIGS["K80_1X"],
worker_config=None,
worker_count=0,
region="us-central1",
args=["1000"],
stream_logs=False,
docker_image_bucket_name=None,
called_from_notebook=False,
)
validate.validate(
entry_point=None,
distribution_strategy=None,
requirements_txt="tests/testdata/requirements.txt",
chief_config=machine_config.COMMON_MACHINE_CONFIGS["K80_1X"],
worker_config=None,
worker_count=0,
region="us-central1",
args=["1000"],
stream_logs=False,
docker_image_bucket_name="abc",
called_from_notebook=True,
)
def test_valid_args(self):
validate.validate(
entry_point='tests/testdata/mnist_example_using_fit.py',
distribution_strategy='auto',
requirements_txt='tests/testdata/requirements.txt',
chief_config=TestValidate.VALID_MACHINE_CONFIG,
worker_config=TestValidate.VALID_MACHINE_CONFIG,
worker_count=1,
region='us-central1',
args=None,
stream_logs=True)
validate.validate(
entry_point='tests/testdata/mnist_example_using_fit.py',
distribution_strategy=None,
requirements_txt='tests/testdata/requirements.txt',
chief_config=TestValidate.VALID_MACHINE_CONFIG,
worker_config=None,
worker_count=0,
region='us-central1',
args=['1000'],
stream_logs=False)
def test_invalid_entry_point(self):
with pytest.raises(ValueError, match=r'Invalid `entry_point`'):
validate.validate(
entry_point='/mnist_example_using_fit.py',
distribution_strategy='auto',
requirements_txt='tests/testdata/requirements.txt',
chief_config=TestValidate.VALID_MACHINE_CONFIG,
worker_config=TestValidate.VALID_MACHINE_CONFIG,
worker_count=1,
region='us-central1',
args=None,
stream_logs=True)
with pytest.raises(ValueError, match=r'Invalid `entry_point`'):
validate.validate(
entry_point='/mnist_example_using_fit.txt',
distribution_strategy='auto',
requirements_txt='tests/testdata/requirements.txt',
chief_config=TestValidate.VALID_MACHINE_CONFIG,
worker_config=TestValidate.VALID_MACHINE_CONFIG,
worker_count=1,
region='us-central1',
args=None,
stream_logs=True)
