File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -210,18 +210,11 @@ def _inplace_fix_subset_categorical_obs(subset_adata: AnnData, original_adata: A
210210 """
211211 if not hasattr (subset_adata , "obs" ) or not hasattr (original_adata , "obs" ):
212212 return
213- # Handle lazy tables (Dataset2D) vs eager tables (DataFrame)
214- if isinstance (subset_adata .obs , pd .DataFrame ):
215- obs = pd .DataFrame (subset_adata .obs )
216- else :
217- # Lazy AnnData uses Dataset2D which needs to_memory() to convert properly
218- obs = subset_adata .obs .to_memory ()
219-
220- # Also handle lazy original_adata.obs
221- if isinstance (original_adata .obs , pd .DataFrame ):
222- original_obs = original_adata .obs
223- else :
224- original_obs = original_adata .obs .to_memory ()
213+ # Handle lazy tables (Dataset2D vs DataFrame). Lazy AnnData uses Dataset2D which needs to_memory()
214+ obs = pd .DataFrame (subset_adata .obs ) if isinstance (subset_adata .obs , pd .DataFrame ) else subset_adata .obs .to_memory ()
215+ original_obs = (
216+ original_adata .obs if isinstance (original_adata .obs , pd .DataFrame ) else original_adata .obs .to_memory ()
217+ )
225218
226219 for column in obs .columns :
227220 is_categorical = isinstance (obs [column ].dtype , pd .CategoricalDtype )
You can’t perform that action at this time.
0 commit comments