def setup(self,
runtime_env: RuntimeEnv,
context: RuntimeEnvContext,
logger: Optional[logging.Logger] = default_logger):
if not runtime_env.working_dir():
return
logger.info(f"Setup working dir for {runtime_env.working_dir()}")
working_dir = download_and_unpack_package(runtime_env.working_dir(),
self._resources_dir,
logger=logger)
context.command_prefix += [f"cd {working_dir}"]
# Insert the working_dir as the first entry in PYTHONPATH. This is
# compatible with users providing their own PYTHONPATH in env_vars.
python_path = working_dir
if "PYTHONPATH" in context.env_vars:
python_path += os.pathsep + context.env_vars["PYTHONPATH"]
context.env_vars["PYTHONPATH"] = python_path
def get_uri(self, runtime_env: RuntimeEnv) -> Optional[str]:
working_dir_uri = runtime_env.working_dir()
if working_dir_uri != "":
return working_dir_uri
return None