type check

Author: c-w-feldmann

molpipeline/mol2any/mol2chemprop.py

@@ -2,19 +2,25 @@
 
 # pylint: disable=too-many-arguments
 
-from collections.abc import Iterable
-from typing import Any, Self
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any, Self
 
 import numpy as np
 import numpy.typing as npt
 from chemprop.data import MoleculeDatapoint, MoleculeDataset
-from chemprop.featurizers.base import GraphFeaturizer, VectorFeaturizer
 
 from molpipeline.abstract_pipeline_elements.core import MolToAnyPipelineElement
 from molpipeline.estimators.chemprop.featurizer_wrapper.graph_wrapper import (
     SimpleMoleculeMolGraphFeaturizer,
 )
-from molpipeline.utils.molpipeline_types import RDKitMol
+
+if TYPE_CHECKING:
+    from collections.abc import Iterable
+
+    from chemprop.featurizers.base import GraphFeaturizer, VectorFeaturizer
+
+    from molpipeline.utils.molpipeline_types import RDKitMol
 
 
 class MolToChemprop(MolToAnyPipelineElement):