TYPE_CHECKING blocks where relevant

Author: fsoubelet

sdds/classes.py

@@ -7,11 +7,15 @@
 https://ops.aps.anl.gov/manuals/SDDStoolkit/SDDStoolkitsu2.html
 """
 
+from __future__ import annotations
+
 import logging
 import warnings
-from collections.abc import Iterator
 from dataclasses import dataclass, fields
-from typing import Any, ClassVar
+from typing import TYPE_CHECKING, Any, ClassVar
+
+if TYPE_CHECKING:
+    from collections.abc import Iterator
 
 LOGGER = logging.getLogger(__name__)
 

sdds/reader.py

@@ -6,6 +6,8 @@
 It provides a high-level function to read SDDS files in different formats, and a series of helpers.
 """
 
+from __future__ import annotations
+
 import gzip
 import os
 import pathlib

sdds/writer.py

@@ -6,10 +6,11 @@
 It provides a high-level function to write SDDS files in different formats, and a series of helpers.
 """
 
+from __future__ import annotations
+
 import pathlib
 import struct
-from collections.abc import Iterable
-from typing import IO, Any
+from typing import IO, TYPE_CHECKING, Any
 
 import numpy as np
 
@@ -25,6 +26,9 @@
     get_dtype_str,
 )
 
+if TYPE_CHECKING:
+    from collections.abc import Iterable
+
 
 def write_sdds(sdds_file: SddsFile, output_path: pathlib.Path | str) -> None:
     """