diff --git a/responsibleai_vision/requirements.txt b/responsibleai_vision/requirements.txt index 195230dab0..d555ac1fc1 100644 --- a/responsibleai_vision/requirements.txt +++ b/responsibleai_vision/requirements.txt @@ -6,5 +6,5 @@ scikit-learn>=0.22.1 scipy>=1.4.1 semver~=2.13.0 responsibleai>=0.35.0 -torchmetrics[detection] +torchmetrics vision_explanation_methods \ No newline at end of file