Feature:3974 Add image tag docker setting

Author: Json-Andriopoulos

docs/book/how-to/containerization/containerization.md

@@ -538,6 +538,19 @@ The repository name will be appended to the registry URI of your container regis
 
 If you don't specify a target repository, the default repository name configured in your container registry stack component settings will be used.
 
+### Specifying Image tags
+
+You can control the tag of the generated Docker images using the image tag option:
+
+```python
+from zenml.config import DockerSettings
+
+docker_settings = DockerSettings(image_tag="1.0.0")
+```
+
+Keep in mind that this will be applied to all images built using the DockerSettings object. If there are multiple
+such images, only one of them will keep the tag while the rest will be untagged.
+
 ### Decoupling Code from Builds
 
 To reuse Docker builds while still using your latest code changes, you need to decouple your code from the build. There are two main approaches:

src/zenml/config/docker_settings.py

@@ -122,6 +122,12 @@ class DockerSettings(BaseSettings):
             this image.
             * If a custom `dockerfile` is specified for this settings
             object, this parent image will be ignored.
+        image_tag: Docker image tag to use for the images that get built.
+
+            Additional notes:
+            * This tag will be used for all images built for the pipeline or step
+            (depending on where you define your DockerSettings). If there are multiple
+            such images, only one of them will keep the tag while the rest will be untagged.
         dockerfile: Path to a custom Dockerfile that should be built. Depending
             on the other values you specify in this object, the resulting
             image will be used directly to run your pipeline or ZenML will use
@@ -207,6 +213,7 @@ class DockerSettings(BaseSettings):
     """
 
     parent_image: Optional[str] = None
+    image_tag: Optional[str] = None
     dockerfile: Optional[str] = None
     build_context_root: Optional[str] = None
     parent_image_build_config: Optional[DockerBuildConfig] = None

src/zenml/pipelines/build_utils.py

@@ -374,11 +374,14 @@ def create_pipeline_build(
             dockerfile = images[item_key].dockerfile
             requirements = images[item_key].requirements
         else:
-            tag = snapshot.pipeline_configuration.name
-            if build_config.step_name:
-                tag += f"-{build_config.step_name}"
-            tag += f"-{build_config.key}"
-            tag = docker_utils.sanitize_tag(tag)
+            if build_config.settings.image_tag:
+                tag = build_config.settings.image_tag
+            else:
+                tag = snapshot.pipeline_configuration.name
+                if build_config.step_name:
+                    tag += f"-{build_config.step_name}"
+                tag += f"-{build_config.key}"
+                tag = docker_utils.sanitize_tag(tag)
 
             include_files = build_config.should_include_files(
                 code_repository=code_repository,

tests/unit/pipelines/test_build_utils.py

@@ -212,8 +212,12 @@ def test_build_uses_correct_settings(mocker, empty_pipeline):  # noqa: F811
 def test_building_with_identical_keys_and_settings(mocker):
     """Tests that two build configurations with identical keys and identical
     settings don't lead to two builds."""
-    build_config_1 = BuildConfiguration(key="key", settings=DockerSettings())
-    build_config_2 = BuildConfiguration(key="key", settings=DockerSettings())
+    build_config_1 = BuildConfiguration(
+        key="key", settings=DockerSettings(image_tag="v1")
+    )
+    build_config_2 = BuildConfiguration(
+        key="key", settings=DockerSettings(image_tag="v1")
+    )
 
     mocker.patch.object(
         Stack,
@@ -240,6 +244,8 @@ def test_building_with_identical_keys_and_settings(mocker):
 
     mock_build_docker_image.assert_called_once()
 
+    assert mock_build_docker_image.call_args[1]["tag"] == "v1"
+
 
 def test_building_with_identical_keys_and_different_settings(mocker):
     """Tests that two build configurations with identical keys and different
@@ -301,8 +307,7 @@ def test_building_with_different_keys_and_identical_settings(mocker):
     assert len(build.images) == 2
     assert build.images["key1"].image == "image_name"
     assert build.images["key2"].image == "image_name"
-
-    mock_build_docker_image.assert_called_once()
+    assert mock_build_docker_image.call_args[1]["tag"] == "pipeline-key1"
 
 
 def test_custom_build_verification(