diff options
Diffstat (limited to 'python/client.py')
| -rw-r--r-- | python/client.py | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/python/client.py b/python/client.py index 21eba000..b89a717a 100644 --- a/python/client.py +++ b/python/client.py @@ -10,12 +10,14 @@ from db import dbconn def run(): hostname = socket.gethostname() ET = dbconn('etdb') - if hostname == 'debian': - q = redis.Redis(unix_socket_path='/var/run/redis/redis.sock') - os.environ['OMP_NUM_THREADS'] = '8' - elif hostname == 'gomez': - q = redis.Redis(host='debian') + if hostname in ['debian', 'gomez']: + q = redis.Redis(host='ziggy') os.environ['OMP_NUM_THREADS'] = '4' + if hostname == 'debian': + os.envirion['OMP_NUM_THREADS'] = '8' + elif hostname == 'ziggy': + q = redis.Redis(unix_socket_path='/run/redis/redis.sock') + os.environ['OMP_NUM_THREADS'] = '8' while True: f, args = loads(q.blpop("tasks")[1].decode('utf-8')) if f == 'generate_scenarios': |