diff --git a/DirectMultiStep/Models/Training.py b/DirectMultiStep/Models/Training.py
index e5cec06..a50abf8 100644
--- a/DirectMultiStep/Models/Training.py
+++ b/DirectMultiStep/Models/Training.py
@@ -22,7 +22,7 @@
 
 from typing import Any, Callable, Dict, List, Optional, Tuple, cast
 
-import lightning as pl
+import lightning as pl  # type: ignore
 import numpy as np
 import torch
 import torch.nn as nn
diff --git a/assess_single.py b/assess_single.py
index fb2b9df..6482775 100644
--- a/assess_single.py
+++ b/assess_single.py
@@ -23,7 +23,7 @@
 from pathlib import Path
 from typing import List, Tuple, cast
 
-import lightning as L
+import lightning as L  # type:ignore
 import rdkit.Chem as Chem  # type: ignore
 import torch
 import yaml
diff --git a/train_nosm.py b/train_nosm.py
index 0b58240..dac007f 100644
--- a/train_nosm.py
+++ b/train_nosm.py
@@ -22,7 +22,7 @@
 
 from pathlib import Path
 
-import lightning as L
+import lightning as L  # type: ignore
 import torch
 from lightning.pytorch.callbacks import ModelCheckpoint, RichModelSummary
 
diff --git a/train_wsm.py b/train_wsm.py
index aad7c14..cb3ed11 100644
--- a/train_wsm.py
+++ b/train_wsm.py
@@ -22,7 +22,7 @@
 
 from pathlib import Path
 
-import lightning as L
+import lightning as L  # type: ignore
 import torch
 from lightning.pytorch.callbacks import ModelCheckpoint, RichModelSummary