Add . to the PYTHONPATH in ImageSpec (#2447)

Signed-off-by: Kevin Su <[email protected]>
flyteorg · Jun 3, 2024 · e2847f0 · e2847f0
1 parent 69a3218
commit e2847f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/plugins/flytekit-envd/flytekitplugins/envd/image_builder.py b/plugins/flytekit-envd/flytekitplugins/envd/image_builder.py
@@ -91,7 +91,7 @@ def create_envd_config(image_spec: ImageSpec) -> str:
     run_commands = _create_str_from_package_list(image_spec.commands)
     conda_channels = _create_str_from_package_list(image_spec.conda_channels)
     apt_packages = _create_str_from_package_list(image_spec.apt_packages)
-    env = {"PYTHONPATH": "/root", _F_IMG_ID: image_spec.image_name()}
+    env = {"PYTHONPATH": "/root:", _F_IMG_ID: image_spec.image_name()}
 
     if image_spec.env:
         env.update(image_spec.env)