From 2f9ea732bf3cf7a3c550151159071f1c313912ae Mon Sep 17 00:00:00 2001 From: Zhiyuan Chen Date: Tue, 26 Mar 2024 23:49:19 +0800 Subject: [PATCH] fix imports Signed-off-by: Zhiyuan Chen --- multimolecule/__init__.py | 7 ------- multimolecule/models/rnabert/__init__.py | 2 +- 2 files changed, 1 insertion(+), 8 deletions(-) diff --git a/multimolecule/__init__.py b/multimolecule/__init__.py index 650d06f2..231bb3ad 100644 --- a/multimolecule/__init__.py +++ b/multimolecule/__init__.py @@ -1,10 +1,3 @@ -from transformers import AutoTokenizer - from . import models, tokenizers -from .models import RnaBertConfig -from .tokenizers import RnaTokenizer - -AutoTokenizer.register(RnaBertConfig, RnaTokenizer) - __all__ = ["models", "tokenizers"] diff --git a/multimolecule/models/rnabert/__init__.py b/multimolecule/models/rnabert/__init__.py index 3a9e9ebe..c388aeff 100644 --- a/multimolecule/models/rnabert/__init__.py +++ b/multimolecule/models/rnabert/__init__.py @@ -5,7 +5,7 @@ from .configuration_rnabert import RnaBertConfig from .modeling_rnabert import RnaBertModel -__all__ = ["RnaBertConfig", "RnaBertModel"] +__all__ = ["RnaBertConfig", "RnaBertModel", "RnaTokenizer"] AutoConfig.register("rnabert", RnaBertConfig) AutoModel.register(RnaBertConfig, RnaBertModel)