AttributeError: Can't pickle local object 'get_batch_cell_embeddings..Dataset'

[thinkCodeai]

I am running Tutorial_ZeroShot_Reference_Mapping.ipynb

ref_embed_adata = scg.tasks.embed_data(
adata,
model_dir,
gene_col=gene_col,
batch_size=64,
device = 'cpu',
use_fast_transformer=False
)

and I am getting this error
AttributeError: Can't pickle local object 'get_batch_cell_embeddings..Dataset'

[qingyan16]

Hi, I met the exact same error. I wonder if you have solved this?