minor bug fix

TuftsBCB · Apr 16, 2024 · f90e32f · f90e32f
1 parent 9fd86ca
commit f90e32f
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -11,3 +11,6 @@ view.html
 test.html
 Untitled.ipynb
 demo_mESC_grn.hdf5
+fetal-mac-edlow_grn.ipynb
+fetal-mac-edlow_preprocess.ipynb
+
diff --git a/regdiffusion/__init__.py b/regdiffusion/__init__.py
@@ -3,12 +3,13 @@
 """
 from . import data
 from . import models
+from . import plot
 
 from .logger import LightLogger, load_logger
 from .trainer import RegDiffusionTrainer
 from .grn import GRN, read_hdf5
 from .evaluator import GRNEvaluator
 
-__all__ = ['data', 'models', 'LightLogger', 'load_logger', 
+__all__ = ['data', 'models', 'plot', 'LightLogger', 'load_logger', 
            'RegDiffusionTrainer', 'GRN', 'read_hdf5', 'GRNEvaluator']
 
diff --git a/regdiffusion/plot/pyvis.py b/regdiffusion/plot/pyvis.py
@@ -39,7 +39,7 @@ def plot_pyvis(
             this_node_size = node_size
         else:
             node_shape = 'star' if node in star_genes else 'dot'
-            this_node_size = 2s * node_size if node in star_genes else node_size
+            this_node_size = 2 * node_size if node in star_genes else node_size
         node_color = None if node_color_dict is None else node_color_dict[node]
         g.add_node(node, label=node, size=this_node_size, 
                    shape=node_shape, color=node_color,