Restore psutil and tbX

Signed-off-by: Shah, Karan <[email protected]>
securefederatedai · Dec 10, 2024 · e103d63 · e103d63
1 parent 9ce8983
commit e103d63
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/openfl-workspace/keras_cnn_mnist/requirements.txt b/openfl-workspace/keras_cnn_mnist/requirements.txt
@@ -1,2 +1 @@
-psutil
 tensorflow==2.13
diff --git a/openfl/utilities/logs.py b/openfl/utilities/logs.py
@@ -3,8 +3,6 @@
 
 
 """Logs utilities."""
-
-import json
 import logging
 
 from rich.console import Console

diff --git a/setup.py b/setup.py
@@ -81,6 +81,7 @@ def run(self):
         'click',
         'pyyaml',
         'rich',
+        'psutil',
         'dynaconf',
         'tqdm',
         'numpy',
@@ -90,6 +91,7 @@ def run(self):
         'pandas',
         'scikit-learn',
         'flatten_json',
+        'tensorboardX',
         'protobuf>=4.22,<6.0.0',
         'grpcio>=1.56.2,<1.66.0',
     ],