Update scvi/data/fields/_dataframe_field.py

scverse · Jan 25, 2024 · 611a3af · 611a3af
1 parent 123518f
commit 611a3af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scvi/data/fields/_dataframe_field.py b/scvi/data/fields/_dataframe_field.py
@@ -243,7 +243,7 @@ def get_summary_stats(self, state_registry: dict) -> dict:
         n_categories = len(np.unique(categorical_mapping))
         return {self.count_stat_key: n_categories}
 
-    def view_state_registry(self, state_registry: dict) -> rich.table.Table:
+    def view_state_registry(self, state_registry: dict) -> rich.table.Table | None:
         """View state registry."""
         source_key = state_registry[self.ORIGINAL_ATTR_KEY]
         mapping = state_registry[self.CATEGORICAL_MAPPING_KEY]