Merge pull request #39 from Vastra-Gotalandsregionen/35912-bump-tenso…

…rflow-image 35912: Bump Tensorflow image to new version
Vastra-Gotalandsregionen · Aug 6, 2024 · f1fa63b · f1fa63b
2 parents 8ff39a1 + f24d17d
commit f1fa63b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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