fix import
Browse files- src/mentioned/train.py +1 -1
src/mentioned/train.py
CHANGED
|
@@ -6,7 +6,7 @@ from lightning.pytorch.callbacks import ModelCheckpoint
|
|
| 6 |
from lightning import Trainer
|
| 7 |
|
| 8 |
from mentioned.model import LitMentionDetector, ModelRegistry
|
| 9 |
-
from
|
| 10 |
|
| 11 |
|
| 12 |
def train(
|
|
|
|
| 6 |
from lightning import Trainer
|
| 7 |
|
| 8 |
from mentioned.model import LitMentionDetector, ModelRegistry
|
| 9 |
+
from mentioned.data import DataRegistry
|
| 10 |
|
| 11 |
|
| 12 |
def train(
|