Error when using 'get_normalized_expression'

I got ‘state_registry’ error when using model.get_normalized_expression on ‘CondSCVI’ model.

KeyError Traceback (most recent call last)
in
----> 1 denoised = model.get_normalized_expression(adata_subset, library_size=1e4)
2 denoised.iloc[:5, :5]

3 frames
/usr/local/lib/python3.8/dist-packages/torch/autograd/grad_mode.py in decorate_context(*args, **kwargs)
25 def decorate_context(*args, **kwargs):
26 with self.clone():
—> 27 return func(*args, **kwargs)
28 return cast(F, decorate_context)
29

/usr/local/lib/python3.8/dist-packages/scvi/model/base/_rnamixin.py in get_normalized_expression(self, adata, indices, transform_batch, gene_list, library_size, n_samples, n_samples_overall, batch_size, return_mean, return_numpy)
100 )
101
→ 102 transform_batch = _get_batch_code_from_category(
103 self.get_anndata_manager(adata, required=True), transform_batch
104 )

/usr/local/lib/python3.8/dist-packages/scvi/model/_utils.py in _get_batch_code_from_category(adata_manager, category)
233 category = [category]
234
→ 235 batch_mappings = adata_manager.get_state_registry(
236 REGISTRY_KEYS.BATCH_KEY
237 ).categorical_mapping

/usr/local/lib/python3.8/dist-packages/scvi/data/_manager.py in get_state_registry(self, registry_key)
371
372 return attrdict(
→ 373 self._registry[_constants._FIELD_REGISTRIES_KEY][registry_key][
374 _constants._STATE_REGISTRY_KEY
375 ]

KeyError: ‘state_registry’