add type ignore

pytorch · Dec 4, 2023 · 9dfa237 · 9dfa237
1 parent 948670f
commit 9dfa237
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchvision/prototype/tv_tensors/_label.py b/torchvision/prototype/tv_tensors/_label.py
@@ -48,7 +48,7 @@ def to_categories(self) -> Any:
         if self.categories is None:
             raise RuntimeError("Label does not have categories")
 
-        return tree_map(lambda idx: self.categories[idx], self.tolist())
+        return tree_map(lambda idx: self.categories[idx], self.tolist())  # type: ignore[index]
 
 
 class OneHotLabel(_LabelBase):