FIX-#404: Add psutil as a required dependency for unidist[mpi] (#406)

Signed-off-by: Igoshev, Iaroslav <[email protected]>
modin-project · Nov 20, 2023 · 5b932f9 · 5b932f9
1 parent 6edebf9
commit 5b932f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -40,7 +40,7 @@ def is_pure(self):
 # https://github.com/modin-project/unidist/issues/324
 ray_deps = ["ray[default]>=1.13.0", "pydantic<2"]
 dask_deps = ["dask[complete]>=2.22.0", "distributed>=2.22.0"]
-mpi_deps = ["mpi4py>=3.0.3", "msgpack>=1.0.0"]
+mpi_deps = ["mpi4py>=3.0.3", "msgpack>=1.0.0", "psutil"]
 if sys.version_info[1] < 8:
     mpi_deps += "pickle5"
 all_deps = ray_deps + dask_deps + mpi_deps