From 2a92974ae8234d65ca5d125c781c510cbd030f6c Mon Sep 17 00:00:00 2001
From: Martin Bauer <martin.bauer@fau.de>
Date: Thu, 7 Jun 2018 09:56:51 +0200
Subject: [PATCH] max domain size info fix
---
runhelper/parameterstudy.py | 15 +++++++++++++--
1 file changed, 13 insertions(+), 2 deletions(-)
diff --git a/runhelper/parameterstudy.py b/runhelper/parameterstudy.py
index 5ec9b2d22..6b47846c9 100644
--- a/runhelper/parameterstudy.py
+++ b/runhelper/parameterstudy.py
@@ -221,7 +221,7 @@ class ParameterStudy:
server.handle_request()
def run_client(self, client_name: str="{hostname}_{pid}", server: str='localhost', port: int=8342,
- parameter_update: Optional[ParameterDict] = None) -> None:
+ parameter_update: Optional[ParameterDict] = None, max_time=None) -> None:
"""Start runner client that retrieves configuration from server, runs it and reports results back to server.
Args:
@@ -232,14 +232,22 @@ class ParameterStudy:
parameter_update: Used to override/extend parameters received from the server.
Typical use cases is to set optimization or GPU parameters for some clients to make
some clients simulate on CPU, others on GPU
+ max_time: maximum runtime in seconds: the client runs scenario after scenario, but starts only a new
+ scenario if not more than max_time seconds have passed since this function was called.
+ So the time given here should be the total maximum runtime minus a typical runtime for one setup
"""
from urllib.request import urlopen
from urllib.error import URLError
+ import time
parameter_update = {} if parameter_update is None else parameter_update
url = "http://{}:{}".format(server, port)
client_name = client_name.format(hostname=socket.gethostname(), pid=os.getpid())
+ start_time = time.time()
while True:
try:
+ if max_time is not None and (time.time() - start_time) > max_time:
+ print("Stopping client - maximum time reached")
+ break
http_response = urlopen(url + "/next_scenario",
data=json.dumps({'client_name': client_name}).encode())
scenario = json.loads(http_response.read().decode())
@@ -269,7 +277,7 @@ class ParameterStudy:
self.run_server(a.host, a.port)
def client(a):
- self.run_client(a.client_name, a.host, a.port, json.loads(a.parameter_override))
+ self.run_client(a.client_name, a.host, a.port, json.loads(a.parameter_override), a.max_time)
def local(a):
if a.database:
@@ -303,6 +311,9 @@ class ParameterStudy:
client_parser.add_argument("-P", "--parameter_override", type=str, default="{}",
help="JSON: the parameter dictionary is updated with these parameters. Use this to "
"set host specific options like GPU call parameters. Enclose in \" ")
+ client_parser.add_argument("-t", "--max_time", type=int, default=None,
+ help="If more than this time in seconds has passed, "
+ "the client stops running scenarios.")
client_parser.set_defaults(func=client)
args = parser.parse_args(argv)
--
GitLab