diff --git a/requirements.txt b/requirements.txt index ad9d6bc..49f2153 100644 --- a/requirements.txt +++ b/requirements.txt @@ -7,7 +7,7 @@ pandas>=1.2.0 Pillow PyYAML scipy>=1.2.1 -sklearn +scikit-learn torch tqdm typing_extensions