diff --git a/tests/test_tensorflow_env.py b/tests/test_tensorflow_env.py index ac172a0fb..8ae53d6b6 100644 --- a/tests/test_tensorflow_env.py +++ b/tests/test_tensorflow_env.py @@ -87,7 +87,7 @@ def test_dockerfile(self): lines = list(map(lambda x: x.strip(), fin.readlines())) dockerfile_text = Path(dockerfile_path).read_text() - assert lines[0] == "FROM registry.git.vgregion.se/aiplattform/images/tensorflow:0.3.1" + assert lines[0] == "FROM registry.git.vgregion.se/aiplattform/images/tensorflow:1.0.0" assert lines[-2] == 'WORKDIR /workspace' assert "ADD http://aiav2.vgregion.se/VGC%20Root%20CA%20v2.crt /tmp/vgc_root.der" in lines assert "ADD http://aiav2.vgregion.se/VGC%20Issuing%201%20CA%20v2.crt /tmp/vgc_issuing1.der" in lines diff --git a/{{ cookiecutter.repo_name }}/Dockerfile b/{{ cookiecutter.repo_name }}/Dockerfile index 41350a037..4c5f6e1a4 100644 --- a/{{ cookiecutter.repo_name }}/Dockerfile +++ b/{{ cookiecutter.repo_name }}/Dockerfile @@ -1,4 +1,4 @@ -{% if cookiecutter.image == 'Tensorflow' -%}FROM registry.git.vgregion.se/aiplattform/images/tensorflow:0.3.1 +{% if cookiecutter.image == 'Tensorflow' -%}FROM registry.git.vgregion.se/aiplattform/images/tensorflow:1.0.0 {% elif cookiecutter.image == 'PyTorch' -%}FROM registry.git.vgregion.se/aiplattform/images/pytorch:1.1.1 {% elif cookiecutter.image == 'R' -%}FROM registry.git.vgregion.se/aiplattform/images/r:0.4.1{%- endif %} ENV GID=1000