Fix deprecated np.long cast

analogdevicesinc · Dec 10, 2024 · 802d98e · 802d98e
1 parent 0fa1fb1
commit 802d98e
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git 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
@@ -10,10 +10,6 @@
 from torch import nn
 
 import matplotlib.pyplot as plt
-import seaborn as sns
-
-sns.set_style("white")
-
 
 DECAY_FACTOR = 1