CostaLab · ruthvik07 · May 6, 2025 · May 6, 2025
diff --git a/dist/pycrosstalker-0.1.0-py3-none-any.whl b/dist/pycrosstalker-0.1.0-py3-none-any.whl
diff --git a/dist/pycrosstalker-0.1.0.tar.gz b/dist/pycrosstalker-0.1.0.tar.gz
diff --git a/docs/source/notebooks/output/LR_data.pkl b/docs/source/notebooks/output/LR_data.pkl
diff --git a/docs/source/notebooks/output/LR_data_final.pkl b/docs/source/notebooks/output/LR_data_final.pkl
diff --git a/docs/source/notebooks/pyCrossTalkeR.ipynb b/docs/source/notebooks/pyCrossTalkeR.ipynb
diff --git a/pycrosstalker/plots/plot.py b/pycrosstalker/plots/plot.py
@@ -593,7 +593,10 @@ def gen_sankey(df, cat_cols=[], value_cols='', title='Sankey Diagram'):
 
     vmin = sourceTargetDf['count'].min()
     vmax = sourceTargetDf['count'].max()
-    norm = mcolors.Normalize(vmin=vmin, vmax=vmax)
+    limit = max(abs(vmin), abs(vmax))
+    vcenter = 0
+    norm = mcolors.TwoSlopeNorm(vmin=-limit, vcenter=vcenter, vmax=limit)
+    # norm = mcolors.Normalize(vmin=vmin, vmax=vmax)
     cmap = plt.get_cmap('RdBu_r')
     sourceTargetDf['hex_color'] = sourceTargetDf['count'].apply(lambda x: mcolors.to_hex(cmap(norm(x))))
 

diff --git a/tutorials/output/LR_data.pkl b/tutorials/output/LR_data.pkl
diff --git a/tutorials/output/LR_data_final.pkl b/tutorials/output/LR_data_final.pkl
diff --git a/tutorials/pyCrossTalkeR.ipynb b/tutorials/pyCrossTalkeR.ipynb