From f90e32f79cf5d570dde72f3cd1d5870afb5a3bb0 Mon Sep 17 00:00:00 2001
From: Hao Zhu <haozhu233@gmail.com>
Date: Tue, 16 Apr 2024 11:25:21 -0400
Subject: [PATCH] minor bug fix

---
 .gitignore                 | 3 +++
 regdiffusion/__init__.py   | 3 ++-
 regdiffusion/plot/pyvis.py | 2 +-
 3 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/.gitignore b/.gitignore
index 3eed2ec..70ff568 100644
--- 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
index 46f3627..e54c3ab 100644
--- 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
index 6d50834..5b48d58 100644
--- 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,