bugfixes

Author: atharvas

src/datasmith/docker/orchestrator.py

@@ -367,7 +367,7 @@ def build_repo_sha_image(
     build_res: BuildResult = docker_ctx.build_container_streaming(
         client=client,
         image_name=task.get_image_name(),
-        build_args={"REPO_URL": repo_url, "COMMIT_SHA": task.sha},
+        build_args={"REPO_URL": repo_url, "COMMIT_SHA": task.sha, "ENV_PAYLOAD": task.env_payload},
         probe=False,
         force=False,
         timeout_s=15 * 60,  # 15 minutes

src/datasmith/docker/validation.py

@@ -217,6 +217,7 @@ def validate_one(  # noqa: C901
         build_args={
             "REPO_URL": f"https://www.github.com/{task.owner}/{task.repo}",
             "COMMIT_SHA": task.sha,
+            "ENV_PAYLOAD": task.env_payload or "",
         },
         force=False,
         timeout_s=args.build_timeout,