Rename internal model references to CodonTranslatorModel
Browse files- src/__init__.py +2 -4
src/__init__.py
CHANGED
|
@@ -3,9 +3,7 @@ CodonTranslator training and inference components.
|
|
| 3 |
"""
|
| 4 |
|
| 5 |
from .tokenizer import CodonTokenizer
|
| 6 |
-
from .models import
|
| 7 |
-
CodonGPT,
|
| 8 |
-
)
|
| 9 |
from .trainer import Trainer, TrainingArguments
|
| 10 |
from .sampler import CodonSampler, sample_sequences
|
| 11 |
from .dataset import (
|
|
@@ -19,7 +17,7 @@ __all__ = [
|
|
| 19 |
# Tokenizer
|
| 20 |
"CodonTokenizer",
|
| 21 |
# Models
|
| 22 |
-
"
|
| 23 |
# Training
|
| 24 |
"Trainer",
|
| 25 |
"TrainingArguments",
|
|
|
|
| 3 |
"""
|
| 4 |
|
| 5 |
from .tokenizer import CodonTokenizer
|
| 6 |
+
from .models import CodonTranslatorModel
|
|
|
|
|
|
|
| 7 |
from .trainer import Trainer, TrainingArguments
|
| 8 |
from .sampler import CodonSampler, sample_sequences
|
| 9 |
from .dataset import (
|
|
|
|
| 17 |
# Tokenizer
|
| 18 |
"CodonTokenizer",
|
| 19 |
# Models
|
| 20 |
+
"CodonTranslatorModel",
|
| 21 |
# Training
|
| 22 |
"Trainer",
|
| 23 |
"TrainingArguments",
|