Apply suggestions from code review

Lightning-AI · Jul 11, 2022 · 4cf373e · 4cf373e
1 parent c209f9c
commit 4cf373e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/torchmetrics/utilities/imports.py b/src/torchmetrics/utilities/imports.py
@@ -112,6 +112,7 @@ def _compare_version(package: str, op: Callable, version: str) -> Optional[bool]
 _SCIPY_AVAILABLE: bool = _package_available("scipy")
 _TORCH_FIDELITY_AVAILABLE: bool = _package_available("torch_fidelity")
 _LPIPS_AVAILABLE: bool = _package_available("lpips")
+_PYCOCOTOOLS_AVAILABLE: bool = _package_available("pycocotools")
 _TORCHVISION_AVAILABLE: bool = _package_available("torchvision")
 _TORCHVISION_GREATER_EQUAL_0_8: Optional[bool] = _compare_version("torchvision", operator.ge, "0.8.0")
 _TQDM_AVAILABLE: bool = _package_available("tqdm")