JuliaPy / JuliaPy/PythonCall.jl

Freeze the resolved interpreter for child processes

未關閉
#796 0 則留言 0 個 reaction 已指派 0 人 在 GitHub 檢視
enhancement
主要語言
Julia
星號
1.1k
分支
86
平均合併
1 天 22 小時
30 天內合併 PR
3

描述

When PythonCall resolves its interpreter via CondaPkg, each **child process** that loads
PythonCall re-resolves the *same* CondaPkg environment from scratch. Spawn several at once —
e.g. `Distributed` workers for parallel data loading (MLUtils `DataLoader(...; num_workers=N)`
over a PythonCall-backed dataset) — and they serialize on CondaPkg's lock, spamming
`CondaPkg: Waiting for lock to be freed` and redoing work the parent already did.

PythonCall already has the fix, but gates it behind `CI=true` and flags it as a hack
(`src/C/context.jl`, v0.9.35):

```julia
# HACK: If we are using CondaPkg, prevent child processes from using it by explicitly
# setting the executable. ...
# A better solution may be to use some environment variable to "freeze" CondaPkg in
# child processes.
# Only done when CI=true since it's a hack.
if (get(ENV, "CI", "false") == "true") && (CTX.which === :CondaPkg)
ENV["JULIA_PYTHONCALL_EXE"] = CTX.exe_path::String
end
```

Since `JULIA_PYTHONCALL_EXE` is exactly the "freeze" env var the comment asks for (child
processes inherit `ENV`, and `getpref_exe` reads it), the same line solves the general case.

## Proposal

Set `JULIA_PYTHONCALL_EXE` whenever `CTX.which === :CondaPkg`, not only under `CI` — ideally
behind a preference (e.g. `freeze_child_processes`, default `true`) so it can be disabled.
`get!` already leaves a user-set value untouched.

## Workaround

A downstream package can do this itself after init:

```julia
if PythonCall.C.CTX.which === :CondaPkg
get!(ENV, "JULIA_PYTHONCALL_EXE", PythonCall.python_executable_path()::String)
end
```

I've done this in https://github.com/JuliaGenAI/HuggingFaceDatasets.jl/pull/64

貢獻指南

這個儲存庫沒有索引到貢獻指南

研究方向

Start in src/C/context.jl and trace how getpref_exe and CTX.which handle CondaPkg during child-process startup. Verify the existing CI-gated behavior and preference conventions, then confirm that spawned processes inherit the resolved executable without re-resolving CondaPkg and that a user-set JULIA_PYTHONCALL_EXE remains unchanged.

由索引模型根據 Issue 內容生成。

評估

技術堆疊
python
領域
backend
Issue 類型
缺陷
難度
3/5
預估耗時
1-2 天
活躍度
冷清
描述清晰度
基本清楚
新手友好度
68/100

把新 issue 寄到你的電子郵件信箱

精選適合新手參與的 GitHub issue 摘要。