deleted additional colocated files

Author: lkolluru05

Dockerfile.colocated

@@ -110,7 +110,7 @@ class Config(DockerBundler.Config):
         # Build image asynchronously.
         colocated_image_required: bool = False
         colocated_image_name: str = None
-        colocated_dockerfile: str = None
+        #colocated_dockerfile: str = None
 
 
     @classmethod
@@ -120,7 +120,7 @@ def from_spec(cls, spec: list[str], *, fv: Optional[flags.FlagValues]) -> Docker
         cfg.dockerfile = cfg.dockerfile or gcp_settings("default_dockerfile", required=False, fv=fv)
         cfg.colocated_image_required = cfg.colocated_image_required or gcp_settings("colocated_image_required", required=False, fv=fv)
         cfg.colocated_image_name = cfg.colocated_image_name or gcp_settings("colocated_image_name", required=False, fv=fv)
-        cfg.colocated_dockerfile = cfg.colocated_dockerfile or gcp_settings("colocated_dockerfile", required=False, fv=fv)
+        #cfg.colocated_dockerfile = cfg.colocated_dockerfile or gcp_settings("colocated_dockerfile", required=False, fv=fv)
         return cfg
 
     def _build_and_push(self, *args, **kwargs):
@@ -131,18 +131,18 @@ def _build_and_push(self, *args, **kwargs):
         )
 
         actual_name = cfg.image
-        actual_dockerfile=cfg.dockerfile
+        #actual_dockerfile=cfg.dockerfile
         actual_target=cfg.target
         if bool(cfg.colocated_image_required):
 
-            cfg.dockerfile=cfg.colocated_dockerfile
+            #cfg.dockerfile=cfg.colocated_dockerfile
             cfg.image=cfg.colocated_image_name
-            cfg.target=None
+            cfg.target="colocated-python"
 
             colocated_bundler_class = ColocatedArtifactRegistryBundler(cfg=cfg)
             colocated_image_name = colocated_bundler_class.bundle(tag=cfg.image)
 
-            cfg.dockerfile=actual_dockerfile
+            #cfg.dockerfile=actual_dockerfile
             cfg.image=actual_name
             cfg.target=actual_target
 
@@ -156,7 +156,7 @@ class ColocatedArtifactRegistryBundler(DockerBundler):
     def from_spec(cls, spec: list[str], *, fv: Optional[flags.FlagValues]) -> DockerBundler.Config:
         cfg: ColocatedArtifactRegistryBundler.Config = super().from_spec(spec, fv=fv)
         cfg.repo = cfg.repo or gcp_settings("docker_repo", required=False, fv=fv)
-        cfg.dockerfile = cfg.colocated_dockerfile or gcp_settings("colocated_dockerfile", required=False, fv=fv)
+        cfg.dockerfile = cfg.colocated_dockerfile or gcp_settings("default_dockerfile", required=False, fv=fv)
         return cfg
 
     def _build_and_push(self, *args, **kwargs):

axlearn/cloud/gcp/bundler.py

@@ -13,8 +13,7 @@ axlearn gcp bundle --name=$NAME \
         --bundler_spec=image=tpu \
         --bundler_spec=target=tpu \
 	  --bundler_spec=colocated_image_required=True \
-	--bundler_spec=colocated_image_name=$COLOCATED_NAME \
-	  --bundler_spec=colocated_dockerfile=Dockerfile.colocated
+	--bundler_spec=colocated_image_name=$COLOCATED_NAME