GIT: type ignore import of lightning

batistagroup · May 31, 2024 · 038e955 · 038e955
1 parent 10a9533
commit 038e955
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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