fix style checks

Signed-off-by: Farhan Ahmed <[email protected]>
Trusted-AI · Nov 30, 2023 · f6c32bd · f6c32bd
1 parent eb0af72
commit f6c32bd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/art/estimators/object_detection/pytorch_detection_transformer.py b/art/estimators/object_detection/pytorch_detection_transformer.py
@@ -21,7 +21,7 @@
 | Paper link: https://arxiv.org/abs/2005.12872
 """
 import logging
-from typing import Any, Dict, List, Optional, Tuple, Union, TYPE_CHECKING
+from typing import Dict, List, Optional, Tuple, Union, TYPE_CHECKING
 
 import numpy as np
 
@@ -144,7 +144,7 @@ def _translate_labels(self, labels: List[Dict[str, "torch.Tensor"]]) -> List[Dic
             height = self.input_shape[0]
             width = self.input_shape[1]
 
-        labels_translated: List[Dict[str, torch.Tensor]] = []
+        labels_translated: List[Dict[str, "torch.Tensor"]] = []
 
         for label_dict in labels:
             label_dict_translated = {}