From 5b932f9b4dd85ab2b7c1a9b2400b6f2b19236e51 Mon Sep 17 00:00:00 2001
From: Iaroslav Igoshev <iaroslav.igoshev@intel.com>
Date: Mon, 20 Nov 2023 14:57:28 +0100
Subject: [PATCH] FIX-#404: Add psutil as a required dependency for
 unidist[mpi] (#406)

Signed-off-by: Igoshev, Iaroslav <iaroslav.igoshev@intel.com>
---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 34d9c560..dc2781bf 100644
--- 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