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=None,
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,
)
with pytest.raises(ValueError, match=r"Invalid `entry_point`"):
validate.validate(
entry_point="/mnist_example_using_fit.txt",
distribution_strategy="auto",
requirements_txt=None,
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,
)
def test_invalid_region(self):
with pytest.raises(ValueError, match=r"Invalid `region`"):
validate.validate(
entry_point=None,
distribution_strategy="auto",
requirements_txt=None,
chief_config=machine_config.COMMON_MACHINE_CONFIGS["K80_1X"],
worker_config=machine_config.COMMON_MACHINE_CONFIGS["K80_1X"],
worker_count=1,
region=["us-region-a"],
args=None,
stream_logs=True,
docker_image_bucket_name=None,
called_from_notebook=False,
)
def test_invalid_tpu_accelerator_count(self):
with pytest.raises(ValueError, match=r"Invalid machine configuration"):
validate.validate(
entry_point=None,
distribution_strategy="auto",
requirements_txt=None,
chief_config=machine_config.COMMON_MACHINE_CONFIGS["CPU"],
worker_config=machine_config.MachineConfig(
accelerator_type=machine_config.AcceleratorType.TPU_V3),
worker_count=1,
region="us-central1",
args=None,
stream_logs=True,
docker_image_bucket_name=None,
called_from_notebook=False,
)
def test_valid_args(self):
validate.validate(
entry_point=
"tensorflow_cloud/python/tests/testdata/mnist_example_using_fit.py",
distribution_strategy="auto",
requirements_txt=
"tensorflow_cloud/python/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=
"tensorflow_cloud/python/tests/testdata/mnist_example_using_fit.py",
distribution_strategy=None,
requirements_txt=
"tensorflow_cloud/python/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=
"tensorflow_cloud/python/tests/testdata/mnist_example_using_fit.ipynb",
distribution_strategy=None,
requirements_txt=
"tensorflow_cloud/python/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=
"tensorflow_cloud/python/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,
)
validate.validate(
entry_point=None,
distribution_strategy=None,
requirements_txt=
"tensorflow_cloud/python/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,
job_labels={"a": "b"},
)