From 802d98ef2f6e4c75dad44509608715acfcbc2dac Mon Sep 17 00:00:00 2001
From: Selda Uyanik <selda.uyanik@maximintegrated.com>
Date: Tue, 10 Dec 2024 14:34:14 -0600
Subject: [PATCH] Fix deprecated np.long cast

---
 datasets/cbm_dataframe_parser.py | 2 +-
 utils/autoencoder_eval_utils.py  | 4 ----
 2 files changed, 1 insertion(+), 5 deletions(-)

diff --git a/datasets/cbm_dataframe_parser.py b/datasets/cbm_dataframe_parser.py
index 7991dbc2a..47ec2dc24 100644
--- a/datasets/cbm_dataframe_parser.py
+++ b/datasets/cbm_dataframe_parser.py
@@ -414,7 +414,7 @@ def __getitem__(self, index):
                 lbl = self.transform(lbl)
 
         if not self.label_as_signal:
-            lbl = lbl.astype(np.long)
+            lbl = lbl.astype(np.int64)
         else:
             lbl = lbl.numpy().astype(np.float32)
 
diff --git a/utils/autoencoder_eval_utils.py b/utils/autoencoder_eval_utils.py
index bb2660f8a..4f89e5750 100755
--- a/utils/autoencoder_eval_utils.py
+++ b/utils/autoencoder_eval_utils.py
@@ -10,10 +10,6 @@
 from torch import nn
 
 import matplotlib.pyplot as plt
-import seaborn as sns
-
-sns.set_style("white")
-
 
 DECAY_FACTOR = 1