From 1035a466be70aaea1388d90bee2802fe96b99a6f Mon Sep 17 00:00:00 2001
From: Nels Schimek <nlsschim@uw.edu>
Date: Thu, 18 Apr 2024 16:42:26 -0700
Subject: [PATCH] heatmaps tests.py

---
 diff_classifier/heatmaps.py | 2 +-
 requirements.txt            | 8 ++++----
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/diff_classifier/heatmaps.py b/diff_classifier/heatmaps.py
index bd46881..fee0dd2 100755
--- 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
index 3d813fe..2da7a91 100644
--- 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