chore(deps): Clarify deps versions in requirements.txt

Minor change in docs in `pyproject.toml`.
KarhouTam · Oct 6, 2024 · 894a935 · 894a935
1 parent ddf06bb
commit 894a935
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 64 deletions.
diff --git a/.env/poetry.lock b/.env/poetry.lock
diff --git a/.env/pyproject.toml b/.env/pyproject.toml
@@ -2,7 +2,7 @@
 name = "fl-bench"
 version = "0.2.0"
 description = "Benchmark of federated learning that aim solving image classification tasks."
-authors = ["KarhouTam <karhoutam@qq.com>"]
+authors = ["KarhouTam <karhoutam@qq.com>", "Xinpeng Wang <223015056@link.cuhk.edu.cn>"]
 license = "MIT"
 readme = ["../README.md", "../data/README.md"]
 repository = "https://github.com/KarhouTam/FL-bench"
@@ -16,7 +16,6 @@ priority = "primary"
 python = ">=3.10, <=3.12"
 torch = "2.2.0"
 torchvision = "^0.17.0"
-torchaudio = "^2.2.0"
 rich = "13.7.1"
 numpy = "1.26.4"
 pandas = "^2.2.2"

diff --git a/.env/requirements.txt b/.env/requirements.txt
@@ -1,18 +1,17 @@
 torch~=2.2.0
-rich
-torchvision
-torchaudio
+rich~=13.7.1
+torchvision~=0.17.0
 numpy<2.0
-visdom
-Pillow
-scipy
-pandas
-matplotlib
-faiss-cpu
-scikit-learn
-pynvml
-PyYAML
-ray[default]
-tensorboard
-cvxpy
-hydra-core
+visdom==0.2.4
+Pillow~=10.3.0
+scipy~=1.13.1
+pandas~=2.2.2
+matplotlib~=3.9.0
+faiss-cpu~=1.7.4
+scikit-learn~=1.5.2
+pynvml~=11.5.0
+PyYAML~=6.0.1
+ray[default]~=2.36.1
+tensorboard~=2.17.1
+cvxpy~=1.5.1
+hydra-core~=1.3.2