From c9f4dc62fdeb12ed34c9ae9253634aa29c3386f3 Mon Sep 17 00:00:00 2001
From: Rishi Sharma <rishi.sharma@epfl.ch>
Date: Wed, 24 Nov 2021 11:36:47 +0100
Subject: [PATCH] Fix termination bug TCP
---
testing.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/testing.py b/testing.py
index 0a32f59..dd52c2b 100644
--- a/testing.py
+++ b/testing.py
@@ -39,6 +39,7 @@ if __name__ == "__main__":
args = parser.parse_args()
Path(args.log_dir).mkdir(parents=True, exist_ok=True)
+ log_level = {"INFO": logging.INFO, "DEBUG": logging.DEBUG, "WARNING": logging.WARNING, "ERROR": logging.ERROR, "CRITICAL": logging.CRITICAL}
config = read_ini(args.config_file)
my_config = dict()
@@ -55,5 +56,5 @@ if __name__ == "__main__":
mp.spawn(
fn=Node,
nprocs=procs_per_machine,
- args=[m_id, l, g, my_config, args.iterations, args.log_dir, args.log_level],
+ args=[m_id, l, g, my_config, args.iterations, args.log_dir, log_level[args.log_level]],
)
--
GitLab