kadarakos commited on
Commit
643063e
·
1 Parent(s): 609f1fa

fix import

Browse files
Files changed (1) hide show
  1. 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 mentions.data import DataRegistry
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(