KeyError: "'columns'" of self.adata_manager.get_state_registry(REGISTRY_KEYS.CONT_COVS_KEY)["columns"]

[YiweiNiu]

Report

Hi, I'm using the latest multigrate (5919c42, with scvi v1.3.3) and got the following error when doing query-reference mapping:

Traceback (most recent call last):
  File "/scratch/multigrate/run_mtg_mudata.py", line 178, in <module>
    query, vae_new = query_mapping(adata_q = query, vae = vae)
  File "/scratch/multigrate/run_mtg_mudata.py", line 117, in query_mapping
    vae_new = mtg.model.MultiVAE.load_query_data(adata_q, vae)
  File "/opt/conda/envs/conda_container_env/lib/python3.10/site-packages/multigrate/model/_multivae.py", line 609, in load_query_data
    model = _initialize_model(cls, adata, registry, attr_dict, datamodule=None)
  File "/opt/conda/envs/conda_container_env/lib/python3.10/site-packages/scvi/model/base/_save_load.py", line 146, in _initialize_model
    model = cls(adata, **non_kwargs, **kwargs)
  File "/opt/conda/envs/conda_container_env/lib/python3.10/site-packages/multigrate/model/_multivae.py", line 129, in __init__
    self.adata_manager.get_state_registry(REGISTRY_KEYS.CONT_COVS_KEY)["columns"]
  File "/opt/conda/envs/conda_container_env/lib/python3.10/site-packages/ml_collections/config_dict/config_dict.py", line 958, in __getitem__
    raise KeyError(self._generate_did_you_mean_message(key, str(e)))
KeyError: "'columns'"

Any help would be very much appreciated.

[alitinet]

Hi, thanks a lot for reporting! Should be fixed in #46.