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)