Code fixes

Author: ppizarror

MLStructFP/db/image/_base.py

@@ -6,7 +6,7 @@
 
 __all__ = ['BaseImage', 'TYPE_IMAGE']
 
-from MLStructFP._types import TYPE_CHECKING, List, NumberType, Optional, Tuple
+from MLStructFP._types import TYPE_CHECKING, List, NumberType, Tuple
 from MLStructFP.utils import make_dirs
 
 import math
@@ -51,7 +51,7 @@ def __init__(self, path: str, save_images: bool, image_size_px: int) -> None:
             make_dirs(path)
             assert os.path.isdir(path), f'Path <{path}> does not exist'
 
-        super(ABC, self).__init__()
+        super().__init__()
         self._image_size = image_size_px
         self._images = []
         self._names = []  # List of image names
@@ -66,7 +66,7 @@ def image_shape(self) -> Tuple[int, int]:
         return self._image_size, self._image_size
 
     @abstractmethod
-    def close(self) -> None:
+    def close(self, *args, **kwargs) -> None:
         """
         Close and delete all generated figures.
         """