diff --git a/sdks/python/container/build.gradle b/sdks/python/container/build.gradle
index a3a75da18fec..fe8ecb98046e 100644
--- a/sdks/python/container/build.gradle
+++ b/sdks/python/container/build.gradle
@@ -58,7 +58,7 @@ for(int i=min_python_version; i<=max_python_version; ++i) {
     }
     dependsOn ':sdks:python:container:py' + cur + ':docker'
     // 3.14 does not have dependencies for ML yet.
-    if (project.hasProperty("include-ml") && i < 14) {
+    if (project.hasProperty("include-ml") && cur != "314") {
       dependsOn ':sdks:python:container:ml:push' + cur
     }
     if (project.hasProperty("include-distroless")) {