From 9f42c1f192cf2dcd9f05a2d8374e298aba1ef576 Mon Sep 17 00:00:00 2001 From: Mylon Jones Date: Tue, 24 Jun 2025 09:24:02 -0400 Subject: [PATCH] Added scikit-learn to the example image-classification requirements.txt (#37506) Co-authored-by: Pavel Iakubovskii --- examples/pytorch/image-classification/requirements.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/examples/pytorch/image-classification/requirements.txt b/examples/pytorch/image-classification/requirements.txt index 4926040789..2779c1c6a8 100644 --- a/examples/pytorch/image-classification/requirements.txt +++ b/examples/pytorch/image-classification/requirements.txt @@ -2,4 +2,5 @@ accelerate>=0.12.0 torch>=1.5.0 torchvision>=0.6.0 datasets>=2.14.0 -evaluate \ No newline at end of file +evaluate +scikit-learn \ No newline at end of file