Fixed bug when adata.raw is None

earmingol · Feb 1, 2024 · efb89bd · efb89bd
1 parent 6a6bf41
commit efb89bd
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/sccellfie/preprocessing/prepare_inputs.py b/sccellfie/preprocessing/prepare_inputs.py
@@ -73,7 +73,9 @@ def preprocess_inputs(adata, gpr_info, task_by_gene, rxn_by_gene, task_by_rxn, v
     task_by_rxn = task_by_rxn.loc[(task_by_rxn != 0).any(axis=1)]
 
     adata2 = adata[:, [True if g in genes else False for g in adata.var_names]]
-    adata2.raw = adata.raw.to_adata()[:, [True if g in genes else False for g in adata.var_names]]
+    if hasattr(adata, 'raw'):
+        if adata.raw is not None:
+            adata2.raw = adata.raw.to_adata()[:, [True if g in genes else False for g in adata.var_names]]
 
     if verbose:
         print(f'Shape of new adata object: {adata2.shape}\n'