Put typing imports in TYPE_CHECKING block for performance

Author: stefmolin

.coveragerc

@@ -9,3 +9,4 @@ exclude_lines =
     @(abc\.)?abstractmethod
     if __name__ == .__main__.:
     raise NotImplemented
+    if TYPE_CHECKING:

pyproject.toml

@@ -104,6 +104,7 @@ lint.select = [
   "PTH", # flake8-use-pathlib
   "RUF", # ruff-specific rules
   "SIM", # flake8-simplify
+  "TC",  # flake8-type-checking (performance improvements)
   "TRY", # tryceratops
   "UP",  # pyupgrade
   "W",   # pycodestyle warning

src/data_morph/bounds/bounding_box.py

@@ -2,12 +2,15 @@
 
 from __future__ import annotations
 
-from collections.abc import Iterable
-from numbers import Number
+from typing import TYPE_CHECKING
 
 from ._utils import _validate_2d
 from .interval import Interval
 
+if TYPE_CHECKING:
+    from collections.abc import Iterable
+    from numbers import Number
+
 
 class BoundingBox:
     """

src/data_morph/cli.py

@@ -4,13 +4,16 @@
 
 import argparse
 import sys
-from collections.abc import Sequence
+from typing import TYPE_CHECKING
 
 from . import __version__
 from .data.loader import DataLoader
 from .morpher import DataMorpher
 from .shapes.factory import ShapeFactory
 
+if TYPE_CHECKING:
+    from collections.abc import Sequence
+
 ARG_DEFAULTS = {
     'output_dir': 'morphed_data',
     'decimals': 2,

src/data_morph/data/dataset.py

@@ -3,15 +3,18 @@
 from __future__ import annotations
 
 from numbers import Number
+from typing import TYPE_CHECKING
 
 import matplotlib.pyplot as plt
-import pandas as pd
-from matplotlib.axes import Axes
 
 from ..bounds.bounding_box import BoundingBox
 from ..bounds.interval import Interval
 from ..plotting.style import plot_with_custom_style
 
+if TYPE_CHECKING:
+    import pandas as pd
+    from matplotlib.axes import Axes
+
 
 class Dataset:
     """

src/data_morph/data/loader.py

@@ -4,19 +4,22 @@
 
 from importlib.resources import files
 from itertools import zip_longest
-from numbers import Number
 from pathlib import Path
-from typing import ClassVar
+from typing import TYPE_CHECKING, ClassVar
 
 import matplotlib.pyplot as plt
 import numpy as np
 import pandas as pd
-from matplotlib.axes import Axes
 
 from .. import MAIN_DIR
 from ..plotting.style import plot_with_custom_style
 from .dataset import Dataset
 
+if TYPE_CHECKING:
+    from numbers import Number
+
+    from matplotlib.axes import Axes
+
 
 class DataLoader:
     """

src/data_morph/morpher.py

@@ -5,13 +5,11 @@
 from functools import partial
 from numbers import Number
 from pathlib import Path
+from typing import TYPE_CHECKING
 
 import numpy as np
-import pandas as pd
 import tqdm
 
-from .bounds.bounding_box import BoundingBox
-from .data.dataset import Dataset
 from .data.stats import get_values
 from .plotting.animation import (
     ease_in_out_quadratic,
@@ -22,7 +20,13 @@
     stitch_gif_animation,
 )
 from .plotting.static import plot
-from .shapes.bases.shape import Shape
+
+if TYPE_CHECKING:
+    import pandas as pd
+
+    from .bounds.bounding_box import BoundingBox
+    from .data.dataset import Dataset
+    from .shapes.bases.shape import Shape
 
 
 class DataMorpher:

src/data_morph/plotting/animation.py

@@ -5,11 +5,12 @@
 import math
 from functools import wraps
 from pathlib import Path
-from typing import Callable
+from typing import TYPE_CHECKING, Callable
 
 from PIL import Image
 
-from ..shapes.bases.shape import Shape
+if TYPE_CHECKING:
+    from ..shapes.bases.shape import Shape
 
 
 def stitch_gif_animation(

src/data_morph/plotting/static.py

@@ -2,21 +2,24 @@
 
 from __future__ import annotations
 
-from collections.abc import Iterable
 from functools import partial
-from numbers import Number
 from pathlib import Path
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
 import matplotlib.pyplot as plt
 import numpy as np
-import pandas as pd
-from matplotlib.axes import Axes
 from matplotlib.ticker import EngFormatter
 
 from ..data.stats import get_values
 from .style import plot_with_custom_style
 
+if TYPE_CHECKING:
+    from collections.abc import Iterable
+    from numbers import Number
+
+    import pandas as pd
+    from matplotlib.axes import Axes
+
 
 @plot_with_custom_style
 def plot(

src/data_morph/shapes/bases/line_collection.py

@@ -2,16 +2,20 @@
 
 from __future__ import annotations
 
-from collections.abc import Iterable
-from numbers import Number
+from typing import TYPE_CHECKING
 
 import matplotlib.pyplot as plt
 import numpy as np
-from matplotlib.axes import Axes
 
 from ...plotting.style import plot_with_custom_style
 from .shape import Shape
 
+if TYPE_CHECKING:
+    from collections.abc import Iterable
+    from numbers import Number
+
+    from matplotlib.axes import Axes
+
 
 class LineCollection(Shape):
     """