linting

deckard/base/attack/attack.py

@@ -379,7 +379,7 @@ def __call__(
                         end_timestamp = time()
                     else:
                         raise e
-                
+
             time_dict.update(
                 {
                     "adv_fit_time_per_sample": (end - start) / (len(samples) * 1e9),

deckard/base/model/model.py

@@ -179,6 +179,7 @@ def __call__(self, data: list, model: object, library=None):
                 end_timestamp = time()
             elif "should be the same" in str(e).lower():
                 import torch
+
                 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
                 data[0] = torch.from_numpy(data[0])
                 data[1] = torch.from_numpy(data[1])