2 files changed, 16 insertions, 6 deletions
diff --git a/fdio.infra.ansible/roles/docker_images/templates/docker-compose-sut.yaml.j2 b/fdio.infra.ansible/roles/docker_images/templates/docker-compose-sut.yaml.j2
index b4713d8552..810374527c 100644
--- a/fdio.infra.ansible/roles/docker_images/templates/docker-compose-sut.yaml.j2
+++ b/fdio.infra.ansible/roles/docker_images/templates/docker-compose-sut.yaml.j2
@@ -1,9 +1,10 @@
-version: "3"
 services:
-  numa-0:
+  sut-0:
     build:
       context: "base/"
       dockerfile: "Dockerfile"
+      args:
+        BASE_IMAGE: "ubuntu:24.04"
     cap_add:
       - NET_RAW
     command: ["/usr/sbin/sshd","-D", "-p", "6001"]
@@ -20,10 +21,12 @@ services:
         source: "{{ volume.source }}"
         target: "{{ volume.target }}"
 {% endfor %}
-  numa-1:
+  sut-1:
     build:
       context: "base/"
       dockerfile: "Dockerfile"
+      args:
+        BASE_IMAGE: "ubuntu:24.04"
     cap_add:
       - NET_RAW
     command: ["/usr/sbin/sshd","-D", "-p", "6002"]
diff --git a/fdio.infra.ansible/roles/docker_images/templates/docker-compose-tg.yaml.j2 b/fdio.infra.ansible/roles/docker_images/templates/docker-compose-tg.yaml.j2
index 2cee85e169..590ba76a38 100644
--- a/fdio.infra.ansible/roles/docker_images/templates/docker-compose-tg.yaml.j2
+++ b/fdio.infra.ansible/roles/docker_images/templates/docker-compose-tg.yaml.j2
@@ -1,9 +1,12 @@
-version: "3"
 services:
   tg-0:
     build:
-      context: "base/"
+      context: "tg/"
       dockerfile: "Dockerfile"
+      args:
+        BASE_IMAGE: "ubuntu:22.04"
+        TREX_VERSION: "3.03"
+        IPERF_VERSION: "3.7"
     command: ["/usr/sbin/sshd","-D", "-p", "6001"]
     expose:
       - "6001"
@@ -20,8 +23,12 @@ services:
 {% endfor %}
   tg-1:
     build:
-      context: "base/"
+      context: "tg/"
       dockerfile: "Dockerfile"
+      args:
+        BASE_IMAGE: "ubuntu:22.04"
+        TREX_VERSION: "3.03"
+        IPERF_VERSION: "3.7"
     command: ["/usr/sbin/sshd","-D", "-p", "6002"]
     expose:
       - "6002"