added gene_adata to list of adatas to assign colors to

mortazavilab · Jan 18, 2023 · 0acbb9c · 0acbb9c
1 parent 325bc11
commit 0acbb9c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/swan_vis/swangraph.py b/swan_vis/swangraph.py
@@ -2306,6 +2306,8 @@ def set_metadata_colors(self, obs_col, cmap):
 		self.tss_adata.uns['{}_dict'] = cmap
 		self.tes_adata.uns['{}_colors'.format(obs_col)] = sample_colors
 		self.tes_adata.uns['{}_dict'] = cmap
+		self.gene_adata.uns['{}_colors'.format(obs_col)] = sample_colors
+		self.gene_adata.uns['{}_dict'] = cmap
 
 	def plot_graph(self, gid,
 				   indicate_dataset=False,