[DRAFT] Add Remote Python Sidecar

src/xpk/commands/workload.py

@@ -254,6 +254,28 @@
               volumeMounts:
               - mountPath: /tmp
                 name: shared-tmp
+            initContainers:
+            # TODO(sujinesh): We should make this optional and only part of the
+            # workload if the user provides the image/enables remote python.
+            - args:
+              name: remote-python-sidecar
+              image: {args.remote_python_sidecar_image}
+              imagePullPolicy: Always
+              command:
+              - "bash"
+              - "-c"
+              - |
+                python start_remote_python.py
+              securityContext:
+                privileged: true
+              volumeMounts:
+              - mountPath: /tmp
+                name: shared-tmp
+              ports:
+                - containerPort: 50051
+                env:
+                - name: GRPC_SERVER_ADDRESS
+                  value: "0.0.0.0:50051"
             nodeSelector:
               {accelerator_label}
               {machine_label}

src/xpk/parser/workload.py

@@ -503,6 +503,14 @@ def add_shared_workload_create_optional_arguments(args_parsers):
             ' event or deletion request.Defaults to 30 seconds.'
         ),
     )
+    custom_parser.add_argument(
+        '--remote-python-sidecar-image',
+        type=str,
+        default='TempImageLocation',
+        help=(
+            'Remote Python sidecar image to use.'
+        ),
+    )
     custom_parser.add_argument(
         '--enable-debug-logs',
         action='store_true',