From 5494e4921562bad9abbdf7efa8b8ebac8029c6bc Mon Sep 17 00:00:00 2001 From: Jack Q Date: Wed, 9 Oct 2024 15:07:46 +0800 Subject: [PATCH] fix python home discovery (#424) --- taskweaver/ces/environment.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/taskweaver/ces/environment.py b/taskweaver/ces/environment.py index 2eb5adc1..ae7c21d2 100644 --- a/taskweaver/ces/environment.py +++ b/taskweaver/ces/environment.py @@ -9,7 +9,7 @@ from dataclasses import dataclass, field from typing import Any, Dict, List, Literal, Optional, Union -from jupyter_client import BlockingKernelClient +from jupyter_client.blocking.client import BlockingKernelClient from jupyter_client.kernelspec import KernelSpec, KernelSpecManager from jupyter_client.manager import KernelManager from jupyter_client.multikernelmanager import MultiKernelManager @@ -187,13 +187,14 @@ def start_session( os.makedirs(cwd, exist_ok=True) if self.mode == EnvMode.Local: - # set python home from current python environment - python_home = os.path.sep.join(sys.executable.split(os.path.sep)[:-2]) + import site + python_path = os.pathsep.join( [ + # package base directory os.path.realpath(os.path.join(os.path.dirname(__file__), "..", "..")), - os.path.join(python_home, "Lib", "site-packages"), ] + + site.getsitepackages() + sys.path, ) @@ -212,7 +213,6 @@ def start_session( "CONNECTION_FILE": self._get_connection_file(session_id, new_kernel_id), "PATH": os.environ["PATH"], "PYTHONPATH": python_path, - "PYTHONHOME": python_home, }, ) session.kernel_id = self.multi_kernel_manager.start_kernel(