From b4f52c6c03a8ba3a9da3d6975e9af66e6a1b120a Mon Sep 17 00:00:00 2001 From: kmontemayor Date: Fri, 1 Aug 2025 23:35:20 +0000 Subject: [PATCH] Swap back to using docker build from scripts/ --- .../KDD_2025/heterogeneous_walkthrough.ipynb | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/examples/tutorial/KDD_2025/heterogeneous_walkthrough.ipynb b/examples/tutorial/KDD_2025/heterogeneous_walkthrough.ipynb index 8dcb3e2ac..020068d58 100644 --- a/examples/tutorial/KDD_2025/heterogeneous_walkthrough.ipynb +++ b/examples/tutorial/KDD_2025/heterogeneous_walkthrough.ipynb @@ -1427,13 +1427,16 @@ "\n", "# In order for your code changes to be reflected in the docker images, you can build your own docker images.\n", "# We have a script that will build the docker images for you.\n", - "from gigl.orchestration.img_builder import build_and_push_customer_src_images\n", + "from scripts.build_and_push_docker_image import build_and_push_cpu_image, build_and_push_cuda_image, build_and_push_dataflow_image\n", + "\n", "repository = \"gigl-base-images\"\n", "tag = f\"{getpass.getuser()}_{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}\"\n", - "notebook_cpu_image, notebook_cuda_image, notebook_dataflow_image = build_and_push_customer_src_images(\n", - " context_path=\".\",\n", - " export_docker_artifact_registry=f\"us-central1-docker.pkg.dev/{resource_config.project}/{repository}\",\n", - ")" + "notebook_cpu_image = f\"us-central1-docker.pkg.dev/{resource_config.project}/{repository}/src-cuda:{tag}\"\n", + "notebook_cuda_image = f\"us-central1-docker.pkg.dev/{resource_config.project}/{repository}/src-cuda:{tag}\"\n", + "notebook_dataflow_image = f\"us-central1-docker.pkg.dev/{resource_config.project}/{repository}/src-dataflow:{tag}\"\n", + "build_and_push_cpu_image(notebook_cpu_image)\n", + "build_and_push_cuda_image(notebook_cuda_image)\n", + "build_and_push_dataflow_image(notebook_dataflow_image)\n" ] }, {