diff --git a/python/fate_test/scripts/benchmark_cli.py b/python/fate_test/scripts/benchmark_cli.py index 365484a..d3d585e 100644 --- a/python/fate_test/scripts/benchmark_cli.py +++ b/python/fate_test/scripts/benchmark_cli.py @@ -116,7 +116,7 @@ def _run_benchmark_pairs(config: Config, suite: BenchmarkSuite, tol: float, name # pipeline demo goes here pair_n = len(suite.pairs) fate_base = config.fate_base - PYTHONPATH = os.environ.get('PYTHONPATH') + ":" + os.path.join(fate_base, "python") + PYTHONPATH = os.environ.get('PYTHONPATH', default="") + ":" + os.path.join(fate_base, "python") os.environ['PYTHONPATH'] = PYTHONPATH for i, pair in enumerate(suite.pairs): echo.echo(f"Running [{i + 1}/{pair_n}] group: {pair.pair_name}") diff --git a/python/fate_test/scripts/performance_cli.py b/python/fate_test/scripts/performance_cli.py index b9f7ab6..93e6423 100644 --- a/python/fate_test/scripts/performance_cli.py +++ b/python/fate_test/scripts/performance_cli.py @@ -166,7 +166,7 @@ def _run_performance_jobs(config: Config, suite: PerformanceSuite, namespace: st # pipeline demo goes here job_n = len(suite.pipeline_jobs) fate_base = config.fate_base - PYTHONPATH = os.environ.get('PYTHONPATH') + ":" + os.path.join(fate_base, "python") + PYTHONPATH = os.environ.get('PYTHONPATH', default="") + ":" + os.path.join(fate_base, "python") os.environ['PYTHONPATH'] = PYTHONPATH job_time_history = {} for j, job in enumerate(suite.pipeline_jobs):