alegendaryfish commited on
Commit
6ef331a
·
verified ·
1 Parent(s): b2f9369

Rename internal model references to CodonTranslatorModel

Browse files
Files changed (1) hide show
  1. 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
- "CodonGPT",
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",