refactor: Removed unused imports, variables and wrappers (#168)

* refactor: Removed unused imports

* refactor: Removed unused variable

* refactor: Removed unnecessary wrapper

holocron/utils/data/collate.py

@@ -3,7 +3,7 @@
 # This program is licensed under the Apache License version 2.
 # See LICENSE or go to <https://www.apache.org/licenses/LICENSE-2.0.txt> for full license details.
 
-from typing import List, Tuple
+from typing import Tuple
 
 import numpy as np
 import torch

references/segmentation/transforms.py

@@ -77,7 +77,6 @@ def __init__(self, prob):
 
     def __call__(self, image, target):
         if random.random() < self.prob:
-            _, width = image.size
             image = F.hflip(image)
             # Flip the segmentation
             target = F.hflip(target)

scripts/eval_latency.py

@@ -36,8 +36,7 @@ def main(args):
 
     # Warmup
     for _ in range(10):
-        with torch.no_grad():
-            _ = model(img_tensor)
+        _ = model(img_tensor)
 
     timings = []
 

tests/test_nn_loss.py

@@ -1,6 +1,6 @@
 import torch
 from torch.nn import CrossEntropyLoss, Linear
-from torch.nn.functional import cross_entropy, log_softmax
+from torch.nn.functional import cross_entropy
 
 from holocron import nn
 from holocron.nn import functional as F

tests/test_requirements.py

@@ -1,5 +1,4 @@
 import re
-from datetime import datetime
 from pathlib import Path