Feat: Update DestVI V2 model

CHANGELOG.md

@@ -14,6 +14,8 @@ to [Semantic Versioning]. Full commit history is available in the
 
 #### Changed
 
+- Update model {class}`scvi.model.DestVI` with fine cell-type classifier {pr}`3380`.
+
 #### Removed
 
 ### 1.3.3 (2025-07-23)

src/scvi/dataloaders/_data_splitting.py

@@ -413,7 +413,7 @@ def __init__(
             labels_state_registry.original_key,
             mod_key=getattr(self.adata_manager.data_registry.labels, "mod_key", None),
         ).ravel()
-        self.unlabeled_category = labels_state_registry.unlabeled_category
+        self.unlabeled_category = getattr(labels_state_registry, "unlabeled_category", None)
         self._unlabeled_indices = np.argwhere(labels == self.unlabeled_category).ravel()
         self._labeled_indices = np.argwhere(labels != self.unlabeled_category).ravel()
 

src/scvi/dataloaders/_semi_dataloader.py

@@ -66,7 +66,7 @@ def __init__(
         # save a nested list of the indices per labeled category
         self.labeled_locs = []
         for label in np.unique(labels):
-            if label != labels_state_registry.unlabeled_category:
+            if label != getattr(labels_state_registry, "unlabeled_category", None):
                 label_loc_idx = np.where(labels[indices] == label)[0]
                 label_loc = self.indices[label_loc_idx]
                 self.labeled_locs.append(label_loc)

src/scvi/external/resolvi/_model.py

@@ -657,7 +657,7 @@ def _set_indices_and_labels(self):
         """Set indices for labeled and unlabeled cells."""
         labels_state_registry = self.adata_manager.get_state_registry(REGISTRY_KEYS.LABELS_KEY)
         self.original_label_key = labels_state_registry.original_key
-        self.unlabeled_category_ = labels_state_registry.unlabeled_category
+        self.unlabeled_category_ = getattr(labels_state_registry, "unlabeled_category", None)
 
         labels = get_anndata_attribute(
             self.adata,