removing pdb

modified: torchsig/utils/visualize.py
TorchDSP · Sep 5, 2024 · 53f56d6 · 53f56d6
1 parent 4eb5854
commit 53f56d6
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/torchsig/utils/visualize.py b/torchsig/utils/visualize.py
@@ -8,7 +8,6 @@
 import numpy as np
 import pywt
 import torch
-import pdbgit 
 
 
 class Visualizer:
@@ -923,7 +922,6 @@ def mask_class_to_outline(tensor: np.ndarray) -> Tuple[List[List[int]], List[Any
     for idx in range(batch_size):
         label = tensor[idx].numpy()
         class_idx_curr = []
-        pdb.set_trace()
         for individual_burst_idx in range(label.shape[0]):
             if np.count_nonzero(label[individual_burst_idx]) > 0:
                 class_idx_curr.append(individual_burst_idx)