moved swan import again

mortazavilab · Jul 30, 2024 · c20f51e · c20f51e
1 parent b2748b5
commit c20f51e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cerberus/cerberusannotation.py b/cerberus/cerberusannotation.py
@@ -215,9 +215,10 @@ def add_sg_info(self, df, sg=None):
         Returns:
             df (pandas DataFrame): DF with SwanGraph t_df info added
         """
-        import swan_vis as swan
 
         if sg is not None:
+            import swan_vis as swan
+
             temp = sg.t_df[['gid', 'gname']].copy(deep=True)
             df.rename({'gene_id': 'gid'}, axis=1, inplace=True)
             temp.rename({'gid':'gene_id'}, axis=1, inplace=True)