fixed mypy types

model/train.py

@@ -13,11 +13,11 @@ from ignite.handlers.param_scheduler import create_lr_scheduler_with_warmup
 from ignite.handlers.tensorboard_logger import TensorboardLogger
 from ignite.handlers.tqdm_logger import ProgressBar
 from ignite.metrics import Average, Loss, RunningAverage
-from src.utils import VCDADataset, get_class
 from torch import nn
 from torch.optim.lr_scheduler import ExponentialLR
 from torch.utils.data import DataLoader
 
+from model.src.utils import VCDADataset, get_class
 from shared.data_store import connect_minio
 
 
@@ -90,7 +90,7 @@ val_metrics = {
 }
 evaluator = create_supervised_evaluator(
     model=model,
-    metrics=val_metrics,
+    metrics=val_metrics,  # type: ignore
     device=device,
 )
 ProgressBar(desc='Val', ncols=80).attach(evaluator)