diff --git a/docker-compose.yml b/docker-compose.yml
index 5bb710d..aa202b1 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -71,6 +71,8 @@ services:
     build:
       context: ./test_experiment
       dockerfile: Dockerfile
+      platforms:
+        - "linux/amd64"
     depends_on:
       - "mlflow"
     environment: