From 5a9c65cb54ab1ccb04242e10d59485fd86a48901 Mon Sep 17 00:00:00 2001
From: Stephen Beckstrom-Sternberg <Stephen.Beckstrom-Sternberg@azdhs.gov>
Date: Wed, 15 May 2024 12:00:00 -0700
Subject: [PATCH] Update heatcluster.py

Changed df.map to df.apply for polars dataframe
---
 heatcluster.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/heatcluster.py b/heatcluster.py
index b7e8ef9..d84cdff 100755
--- a/heatcluster.py
+++ b/heatcluster.py
@@ -109,7 +109,7 @@ def determine_heatmap_size(df, SNPmatrix):
     ##df = df.sort("0").collect()
     ##df = df.drop_nulls().collect()
 
-    labels = df.map(lambda v: '10K+' if v > 10000 else v)
+    labels = df.apply(lambda v: '10K+' if v > 10000 else v)
     """
     Save sorted csv SNP matrix
     """