Fix staging Dockerfile-based templates on release (#1933)

Signed-off-by: Jeffrey Kinard <[email protected]>
GoogleCloudPlatform · Oct 14, 2024 · 7e44934 · 7e44934
1 parent cca2a24
commit 7e44934
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/...maven-plugin/src/main/java/com/google/cloud/teleport/plugin/maven/TemplatesStageMojo.java b/...maven-plugin/src/main/java/com/google/cloud/teleport/plugin/maven/TemplatesStageMojo.java
@@ -210,10 +210,6 @@ public void execute() throws MojoExecutionException {
       stagePrefix = stagePrefix.substring(0, stagePrefix.length() - 1);
     }
 
-    if (beamVersion == null || beamVersion.isEmpty()) {
-      beamVersion = project.getProperties().getProperty("beam-python.version");
-    }
-
     try {
       URLClassLoader loader = buildClassloader();
 
@@ -385,6 +381,10 @@ protected String stageFlexTemplate(
         generateFlexTemplateImagePath(
             containerName, projectId, artifactRegion, artifactRegistry, stagePrefix));
 
+    if (beamVersion == null || beamVersion.isEmpty()) {
+      beamVersion = project.getProperties().getProperty("beam-python.version");
+    }
+
     String currentTemplateName = definition.getTemplateAnnotation().name();
     TemplateSpecsGenerator generator = new TemplateSpecsGenerator();