From 262394e112aa66f17797089a5a8bcb51e0eb4bdd Mon Sep 17 00:00:00 2001
From: Matthias <xmatthias@outlook.com>
Date: Thu, 25 Feb 2021 19:24:30 +0100
Subject: [PATCH] Add psutils to support OOM Gracefull shutdown

closes #4436, #4439 #3990
---
 requirements-hyperopt.txt | 1 +
 setup.py                  | 1 +
 2 files changed, 2 insertions(+)

diff --git a/requirements-hyperopt.txt b/requirements-hyperopt.txt
index 8e87a434c..8cdb6fd28 100644
--- a/requirements-hyperopt.txt
+++ b/requirements-hyperopt.txt
@@ -7,4 +7,5 @@ scikit-learn==0.24.1
 scikit-optimize==0.8.1
 filelock==3.0.12
 joblib==1.0.1
+psutil==5.8.0
 progressbar2==3.53.1
diff --git a/setup.py b/setup.py
index 148803cd6..118bc8485 100644
--- a/setup.py
+++ b/setup.py
@@ -28,6 +28,7 @@ hyperopt = [
     'filelock',
     'joblib',
     'progressbar2',
+    'psutil',
     ]
 
 develop = [