Add mapping to lit model

1 files changed, 1 insertions, 1 deletions

diff --git a/training/run.py b/training/run.py
index 06426e0..0e17b4e 100644
--- a/training/run.py
+++ b/training/run.py
@@ -14,7 +14,6 @@ from pytorch_lightning import (
 from pytorch_lightning.loggers import LightningLoggerBase
 from torch import nn
 from torchinfo import summary
-from text_recognizer.data.base_mapping import AbstractMapping
 
 import utils
 
@@ -40,6 +39,7 @@ def run(config: DictConfig) -> Optional[float]:
     model: LightningModule = hydra.utils.instantiate(
         config.model,
         network=network,
+        mapping=datamodule.mapping,
         loss_fn=loss_fn,
         optimizer_configs=config.optimizers,
         lr_scheduler_configs=config.lr_schedulers,