kernel always restart

[dhtc]

Hi, I have met such a strange bug. ​The kernel always restarted when I ran sc.pp.neighbor after importing scvi package.

# code for illustration
# without importing scvi package, it would run correctly.
import scanpy as sc
adata = sc.read('test.h5ad')
sc.pp.neighbors(adata)

# if I import scvi package, the kernel will always restart.
import scanpy as sc
import scvi
adata = sc.read('test.h5ad')
sc.pp.neighbors(adata) # kernel will restart suddenly

Jupyter lab will restart, and it said " AsyncIOLoopKernelRestarter: restarting kernel (1/5), keep random ports" in log file.

Versions:

VERSION 0.17.1

[adamgayoso]

This suggests some sort of memory leak. Would you be able to try with Google Colab?

[adamgayoso]

Closing due to inactivity.