diff --git a/art/estimators/object_detection/pytorch_detection_transformer.py b/art/estimators/object_detection/pytorch_detection_transformer.py index abe92d6e17..79d6a6c23d 100644 --- 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 = {}