heatmaps tests.py

Nance-Lab · Apr 18, 2024 · 1035a46 · 1035a46
1 parent 356104d
commit 1035a46
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/diff_classifier/heatmaps.py b/diff_classifier/heatmaps.py
@@ -164,7 +164,7 @@ def plot_heatmap(prefix, feature='asymmetry1', vmin=0, vmax=1, resolution=512, r
     regions, vertices = voronoi_finite_polygons_2d(vor)
     my_map = cm.get_cmap('viridis')
     norm = mpl.colors.Normalize(t_min, t_max, clip=True)
-    mapper = cm.ScalarMappable(norm=norm, cmap=cm.viridis)
+    mapper = cm.ScalarMappable(norm=norm, cmap=my_map)
 
     test = 0
     p2 = 0

diff --git a/requirements.txt b/requirements.txt
@@ -1,6 +1,6 @@
-pytest
-pandas
-scikit-learn
+pytest==7.3.1
+pandas==1.5.3
+scikit-learn==1.2.2
 boto3
 shapely
 numpydoc
@@ -13,4 +13,4 @@ notebook
 numpy
 pyimagej
 scikit-image
-
+matplotlib==3.7.1