fix: remove unnecessary import

Kohulan · Dec 8, 2023 · 10a0b58 · 10a0b58
1 parent 7206849
commit 10a0b58
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/decimer_segmentation/complete_structure.py b/decimer_segmentation/complete_structure.py
@@ -1,5 +1,4 @@
 import cv2
-import math
 import numpy as np
 import matplotlib.pyplot as plt
 import itertools
@@ -271,7 +270,8 @@ def complete_structure_mask(
         image_array (np.array): input image
         mask_array (np.array): shape: y, x, n where n is the amount of masks
         max_depiction_size (Tuple[int, int]): height, width
-        debug (bool, optional): More verbose if True. Defaults to False.
+        debug (bool, optional): You get visualisations in a Jupyter Notebook if True.
+            Defaults to False.
 
     Returns:
         np.array: expanded mask array