refactor(baselines) Remove pillow package from FedMeta baseline (#…

…4773)
adap · Jan 6, 2025 · 004af89 · 004af89
1 parent d3d52e5
commit 004af89
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/baselines/fedmeta/README.md b/baselines/fedmeta/README.md
@@ -74,6 +74,10 @@ poetry shell
 ## Running the Experiments
 
 **Download Dataset:** Go [LEAF: A Benchmark for Federated Settings](https://github.com/TalwalkarLab/leaf) and Use the command below! You can download dataset (FEMNIST and SHAKESPEARE). 
+
+> \[!NOTE\]
+> Since the leaf code was written a while ago, you might need to downgrade your pillow package to `9.5.0` before you can run the processing scripts below.
+
 ```bash
 # clone LEAF repo
 git clone https://github.com/TalwalkarLab/leaf.git

diff --git a/baselines/fedmeta/pyproject.toml b/baselines/fedmeta/pyproject.toml
@@ -44,7 +44,6 @@ matplotlib = "3.7.1"
 scikit-learn = "1.3.1"
 torch = { url = "https://download.pytorch.org/whl/cu117/torch-2.0.1%2Bcu117-cp310-cp310-linux_x86_64.whl" }
 torchvision = { url = "https://download.pytorch.org/whl/cu117/torchvision-0.15.2%2Bcu117-cp310-cp310-linux_x86_64.whl" }
-pillow = "9.5.0" # needed <10.0.0 for LEAF repo scripts
 
 [tool.poetry.dev-dependencies]
 isort = "==5.13.2"
@@ -55,7 +54,7 @@ pylint = "==2.8.2"
 flake8 = "==3.9.2"
 pytest = "==6.2.4"
 pytest-watch = "==4.2.0"
-ruff = "==0.0.272"
+ruff = "==0.4.5"
 types-requests = "==2.27.7"
 virtualenv = "==20.21.0"