Update jabba.py

nla-group · Jul 8, 2024 · 4098190 · 4098190
1 parent 2e8ac4f
commit 4098190
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/fABBA/jabba/jabba.py b/fABBA/jabba/jabba.py
@@ -685,10 +685,10 @@ def recast_shape(self, reconstruct_list):
         fixed_len = self.recap_shape[1] * self.recap_shape[2]
 
         if fixed_len > np.max(size_list):
-            warnings.warn('The reconstructed shape has been expanded.', ShapeWarning)
+            warnings.warn('The reconstructed shape has been expanded.')
 
         elif fixed_len < np.max(size_list):
-            warnings.warn('The reconstructed shape has been squeezed.', ShapeWarning)
+            warnings.warn('The reconstructed shape has been squeezed.')
 
         org_size = len(reconstruct_list)
 
@@ -1169,10 +1169,6 @@ def fillna(series, method='ffill'):
 
 
 
-class ShapeWarning(EncodingWarning):
-    pass
-
-
 def zip_longest(*iterables, fillvalue=None):
     # zip_longest('ABCD', 'xy', fillvalue='-') → Ax By C- D-