"prepare_model" function in "Predicting donor-specific cytokine effects on PBMCs" demo get an error #253
Description
Report
Hi,
When I run the example "Predicting donor-specific cytokine effects on PBMCs", an error occurs during the "prepare_model" step. The error message is as follows:
Thanks for your reply.
anndata 0.10.8
cellflow NA
flax 0.9.0
jax 0.6.2
jaxlib 0.6.2
matplotlib 3.9.2
numpy 1.26.4
optax 0.2.5
pandas 2.2.3
rapids_singlecell 0.12.7
scanpy 1.10.3
scipy 1.14.1
seaborn 0.13.2
session_info 1.0.0
torch 2.4.1+cu121
PIL 10.4.0
absl NA
adjustText 1.2.0
anyio NA
array_api_compat 1.8
arrow 1.3.0
asttokens NA
attr 24.2.0
attrs 24.2.0
babel 2.17.0
cachetools 6.1.0
certifi 2024.08.30
cffi 1.17.1
charset_normalizer 3.3.2
chex 0.1.89
cloudpickle 3.1.1
comm 0.2.2
cuda 12.9.0
cudf 25.06.00
cuml 25.06.00
cupy 13.4.1
cupy_backends NA
cupyx NA
cycler 0.12.1
cython_runtime NA
dask 2025.5.0
dateutil 2.9.0.post0
debugpy 1.8.14
decorator 5.2.1
defusedxml 0.7.1
diffrax 0.7.0
distutils 3.12.6
equinox 0.12.2
etils 1.9.4
executing 2.2.0
fastjsonschema NA
fastrlock 0.8.3
filelock 3.16.1
fqdn NA
h5py 3.11.0
huggingface_hub 0.33.0
idna 3.10
igraph 0.11.6
importlib_resources NA
ipykernel 6.29.5
isoduration NA
jaraco NA
jaxopt NA
jaxtyping 0.2.34
jedi 0.19.2
jinja2 3.1.4
joblib 1.4.2
json5 0.12.0
jsonpointer 3.0.0
jsonschema 4.24.0
jsonschema_specifications NA
jupyter_events 0.12.0
jupyter_server 2.16.0
jupyterlab_server 2.27.3
kiwisolver 1.4.7
legacy_api_wrap NA
leidenalg 0.10.2
libcudf 25.06.00
libcuml 25.06.00
libcuvs 25.06.01
libkvikio 25.06.00
libraft 25.06.00
librmm 25.06.00
lineax 0.0.8
llvmlite 0.43.0
markupsafe 2.1.5
matplotlib_inline 0.1.7
ml_dtypes 0.5.0
more_itertools 10.3.0
mpl_toolkits NA
mpmath 1.3.0
msgpack 1.1.0
natsort 8.4.0
nbformat 5.10.4
numba 0.60.0
nvtx NA
opt_einsum v3.3.0
optimistix 0.0.10
ott 0.5.0
overrides NA
packaging 24.1
parso 0.8.4
patsy 0.5.6
pexpect 4.9.0
platformdirs 4.3.8
prometheus_client NA
prompt_toolkit 3.0.51
psutil 7.0.0
ptyprocess 0.7.0
pure_eval 0.2.3
pyarrow 17.0.0
pydev_ipython NA
pydevconsole NA
pydevd 3.2.3
pydevd_file_utils NA
pydevd_plugins NA
pydevd_tracing NA
pygments 2.18.0
pylibcudf NA
pylibraft 25.06.00
pynvjitlink 0.6.0
pynvml NA
pyparsing 3.1.4
pythonjsonlogger NA
pytz 2024.2
rapids_dask_dependency NA
rapids_logger 0.1.1
referencing NA
regex 2.5.148
requests 2.32.3
rfc3339_validator 0.1.4
rfc3986_validator 0.1.1
rich NA
rmm 25.06.00
rpds NA
safetensors 0.5.3
send2trash NA
setuptools 74.1.2
six 1.16.0
sklearn 1.5.1
sniffio 1.3.1
stack_data 0.6.3
statsmodels 0.14.3
sympy 1.13.3
tblib 3.1.0
texttable 1.7.0
threadpoolctl 3.5.0
tlz 0.12.1
tokenizers 0.21.2
toolz 0.12.1
torchgen NA
tornado 6.5.1
tqdm 4.66.5
traitlets 5.14.3
transformers 4.52.4
treelite 4.4.1
triton 3.0.0
typeguard NA
typing_extensions NA
uri_template NA
urllib3 2.2.3
wadler_lindig 0.1.7
wcwidth 0.2.13
webcolors NA
websocket 1.8.0
xarray 2025.6.1
yaml 6.0.2
zmq 27.0.0
IPython 9.3.0
jupyter_client 8.6.3
jupyter_core 5.8.1
jupyterlab 4.4.3
Python 3.12.6 | packaged by conda-forge | (main, Sep 22 2024, 14:16:49) [GCC 13.3.0]
Linux-6.8.0-48-generic-x86_64-with-glibc2.39
Session information updated at 2025-06-26 16:48