Logging of ClusterAutoK() with GPU acceleration #101
Description
Report
The message 'LOCAL_RANK: 0 - CUDA_VISIBLE_DEVICES: [0]' is repeated multiple times (186) when running autok.fit(adata) and I was wondering if there was a way to disable this.
Version information
-----
anndata 0.12.2
cellcharter 0.3.5
lightning 2.5.5
matplotlib 3.10.6
numpy 2.2.6
pandas 2.3.2
scanpy 1.11.4
scvi 1.3.3
session_info v1.0.1
squidpy 1.6.5
torch 2.8.0+cu126
-----
PIL 11.3.0
absl 2.3.1
affine 2.4.0
anyio NA
arrow 1.3.0
asciitree NA
asttokens NA
attr 25.3.0
attrs 25.3.0
babel 2.17.0
brotli 1.1.0
certifi 2025.08.03
cffi 1.17.1
charset_normalizer 3.4.3
chex 0.1.90
click 8.2.1
cloudpickle 3.1.1
comm 0.2.3
crc32c NA
cycler 0.12.1
cython_runtime NA
dask 2024.11.2
dask_image NA
datashader 0.18.2
dateutil 2.9.0.post0
debugpy 1.8.16
decorator 5.2.1
defusedxml 0.7.1
deprecated 1.2.18
docrep 0.3.2
etils 1.13.0
executing 2.2.1
fastjsonschema NA
filelock 3.19.1
flax 0.11.2
fqdn NA
fsspec 2025.9.0
geopandas 1.1.1
gmpy2 2.2.1
h5py 3.14.0
idna 3.10
imageio 2.37.0
importlib_resources NA
ipykernel 6.30.1
ipywidgets 8.1.7
isoduration NA
jaraco NA
jax 0.6.2
jaxlib 0.6.2
jedi 0.19.2
jinja2 3.1.6
joblib 1.5.2
json5 0.12.1
jsonpointer 3.0.0
jsonschema 4.25.1
jsonschema_specifications NA
jupyter_events 0.12.0
jupyter_server 2.17.0
jupyterlab_server 2.27.3
kiwisolver 1.4.9
lark 1.2.2
lazy_loader 0.4
legacy_api_wrap NA
lightning_fabric 2.5.5
lightning_utilities 0.15.2
llvmlite 0.44.0
markupsafe 3.0.2
matplotlib_inline 0.1.7
matplotlib_scalebar 0.9.0
ml_collections 1.1.0
ml_dtypes 0.5.3
more_itertools 10.8.0
mpl_toolkits NA
mpmath 1.3.0
msgpack 1.1.1
mudata 0.3.2
multipledispatch 0.6.0
multiscale_spatial_image 2.0.3
natsort 8.4.0
nbformat 5.10.4
networkx 3.5
numba 0.61.2
numcodecs 0.15.1
numpyro 0.19.0
nvidia NA
ome_zarr NA
opt_einsum 3.4.0
optax 0.2.5
optree 0.17.0
packaging 25.0
param 2.2.1
parso 0.8.5
patsy 1.0.1
pkg_resources NA
platformdirs 4.4.0
prometheus_client NA
prompt_toolkit 3.0.52
psutil 7.0.0
pure_eval 0.2.3
pyarrow 21.0.0
pycparser 2.22
pyct 0.5.0
pydev_ipython NA
pydevconsole NA
pydevd 3.2.3
pydevd_file_utils NA
pydevd_plugins NA
pydevd_tracing NA
pygments 2.19.2
pyparsing 3.2.3
pyproj 3.7.2
pyro 1.9.1
pythonjsonlogger NA
pytorch_lightning 2.5.5
pytz 2025.2
rasterio 1.4.3
referencing NA
requests 2.32.5
rfc3339_validator 0.1.4
rfc3986_validator 0.1.1
rfc3987_syntax NA
rich NA
rpds NA
scipy 1.16.1
seaborn 0.13.2
send2trash NA
setuptools 80.9.0
shapely 2.1.1
simplejson 3.20.1
six 1.17.0
skimage 0.25.2
sklearn 1.7.2
sknw 0.1
sniffio 1.3.1
socks 1.7.1
sparse 0.17.0
spatial_image 1.2.3
spatialdata 0.5.0
spatialdata_plot 0.2.11
stack_data 0.6.3
statsmodels 0.14.5
sympy 1.14.0
testing NA
threadpoolctl 3.6.0
tifffile 2025.9.9
tlz 1.0.0
toolz 1.0.0
torchgen NA
torchgmm NA
torchmetrics 1.8.2
torchvision 0.23.0+cu126
tornado 6.5.2
tqdm 4.67.1
traitlets 5.14.3
triton 3.4.0
typing_extensions NA
uri_template NA
urllib3 2.5.0
validators 0.35.0
wcwidth 0.2.13
webcolors NA
websocket 1.8.0
wrapt 1.17.3
xarray 2025.9.0
xarray_dataclass 3.0.0
xarray_schema 0.0.3
xrspatial 0.4.0
yaml 6.0.2
zarr 2.18.7
zmq 27.1.0
zoneinfo NA
zstandard 0.24.0
-----
IPython 9.5.0
jupyter_client 8.6.3
jupyter_core 5.8.1
jupyterlab 4.4.7
-----
Python 3.12.11 | packaged by conda-forge | (main, Jun 4 2025, 14:45:31) [GCC 13.3.0]
Linux-4.18.0-553.69.1.el8_10.x86_64-x86_64-with-glibc2.28
-----