From 038e95598984a9d160bef05f704d420c9d8aa468 Mon Sep 17 00:00:00 2001 From: Anton Morgunov Date: Fri, 31 May 2024 17:54:35 -0400 Subject: [PATCH] GIT: type ignore import of lightning --- DirectMultiStep/Models/Training.py | 2 +- assess_single.py | 2 +- train_nosm.py | 2 +- train_wsm.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) 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