diff --git a/environment_cpu.yml b/environment_cpu.yml
index c5e781d..383d4a1 100644
--- a/environment_cpu.yml
+++ b/environment_cpu.yml
@@ -28,6 +28,7 @@ dependencies:
   - gcsfs
   - pytest
   - pip:
+      - setuptools
       - datasets
       - einops
       - fsspec