Attempt to fix circular imports

chemprop · Apr 17, 2024 · 275b8cf · 275b8cf
1 parent edf0a92
commit 275b8cf
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/chemprop/featurizers/atom.py b/chemprop/featurizers/atom.py
@@ -4,9 +4,10 @@
 import numpy as np
 from rdkit.Chem.rdchem import Atom, HybridizationType
 
-from chemprop.featurizers.base import VectorFeaturizer
 from chemprop.utils.utils import EnumMapping
 
+from .base import VectorFeaturizer
+
 
 class MultiHotAtomFeaturizer(VectorFeaturizer[Atom]):
     """A :class:`MultiHotAtomFeaturizer` uses a multi-hot encoding to featurize atoms.