imports

Author: ncfrey

src/lobster/transforms/_atom3d_ppi_transforms.py

@@ -6,7 +6,8 @@
 from pandas import DataFrame
 
 from lobster.tokenization._pmlm_tokenizer import PmlmTokenizer
-from lobster.transforms import Transform
+
+from ._transform import Transform
 
 
 def _protein_letters_3to1_wrapper(three_letter: str) -> str:

src/lobster/transforms/_auto_tokenizer_transform.py

@@ -8,7 +8,7 @@
     TruncationStrategy,
 )
 
-from lobster.transforms import Transform
+from ._transform import Transform
 
 
 class AutoTokenizerTransform(Transform):

src/lobster/transforms/_lambda.py

@@ -1,6 +1,6 @@
 from typing import Any, Callable, Dict, Type
 
-from lobster.transforms import Transform
+from ._transform import Transform
 
 
 class Lambda(Transform):