sm/smbinterp
1
0
Fork 0
Code Issues Pull Requests Activity

improved the shepherd and enabled actual task submission

Browse Source
This commit is contained in:
Stephen McQuay 2011-05-18 12:57:28 -06:00
parent 5509ef5b96
commit 2a4a5b7ebe
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
bin/shepherd.py
View File

@ -21,7 +21,7 @@ log = logging.getLogger("interp")
from interp.cluster import QueueManager, get_qs
if __name__ == '__main__':
parser = OptionParser(usage = "%prog [options] <server> <destination vertexes file (pickle)>")
parser = OptionParser(usage = "%prog [options] <server> <destination vertexes file (pickle)> <results pickle>")
parser.add_option("-v", "--verbose",
action="store_true", dest="verbose", default=False,
@ -36,11 +36,11 @@ if __name__ == '__main__':
help="specify how many participants we should wait for (default: %default)")
(options, args) = parser.parse_args()
if len(args) != 2:
if len(args) != 3:
parser.print_usage()
sys.exit(1)
server, dest_verts = args
server, dest_verts, results_p_name = args
dest_verts = pickle.load(open(dest_verts, 'r'))
@ -89,7 +89,7 @@ if __name__ == '__main__':
for X in d:
cur_job = (i, order, extra, X)
results[i] = [ order, extra, X]
# tasksq.put(cur_job)
tasksq.put(cur_job)
i+=1
pbar.update(i)
submit_end = time.time()
@ -114,7 +114,7 @@ if __name__ == '__main__':
receive_end = time.time()
pbar.finish()
pickle.dump(results, open("results.p"))
pickle.dump(results, open(results_p_name, 'w'))
submit = submit_end - submit_start
receive = receive_end - receive_start