pyproject: add psutil dep. for fast_pytorch_kmeans

braun-steven · Nov 11, 2024 · 87551e7 · 87551e7
1 parent b555163
commit 87551e7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -17,6 +17,7 @@ dependencies = [
     "torch~=2.0",
     # "fast_pytorch_kmeans~=0.2.0",
     "fast_pytorch_kmeans@git+https://github.com/DeMoriarty/fast_pytorch_kmeans#egg=1d41c5bda5647e344da3d5432f81f96f6fe21cf6",
+    "psutil~=6.0.0",  # Req. by fast_pytorch_kmeans
     "tqdm~=4.0",
     "scipy~=1.14.0",
     "imageio~=2.36.0"