diff --git a/mmengine/structures/instance_data.py b/mmengine/structures/instance_data.py index 369d445f28..9de326cf25 100644 --- a/mmengine/structures/instance_data.py +++ b/mmengine/structures/instance_data.py @@ -26,7 +26,7 @@ LongTypeTensor = Union[torch.LongTensor, torch.cuda.LongTensor] IndexType: Union[Any] = Union[str, slice, int, list, LongTypeTensor, - BoolTypeTensor, np.ndarray] + BoolTypeTensor, np.ndarray, torch.Tensor] # torch.Tensor is added to support mps device tensor types # Modified from