Applied ruff fixes to code

Signed-off-by: George Araujo <[email protected]>

Author: george-gca

models/srgan.py

@@ -6,7 +6,7 @@
 import torch.nn as nn
 import torch.optim as optim
 from kornia.color import rgb_to_grayscale
-from losses.losses import PSNR, GANLoss, TVLoss, VGGLoss
+from losses.losses import GANLoss, TVLoss, VGGLoss
 from torch.optim.lr_scheduler import StepLR
 from torchvision.utils import make_grid
 

predict.py

@@ -1,7 +1,6 @@
 import argparse
 import inspect
 import logging
-from typing import List
 from pathlib import Path
 
 import numpy as np

train.py

@@ -1,7 +1,6 @@
 import argparse
 import inspect
 import logging
-from typing import List
 from pathlib import Path
 
 import numpy as np