add keepalive

2014-12-03 13:21:30 -08:00
parent 2523288509
commit 7a983a4079
3 changed files with 15 additions and 3 deletions
--- a/src/rabit_master.py
+++ b/src/rabit_master.py
@@ -58,7 +58,7 @@ class SlaveEntry:
        if self.rank >= 0:
            return self.rank
        if self.jobid != 'NULL' and self.jobid in job_map:
-            job_map[self.jobid]
+            return job_map[self.jobid]
        return -1

    def get_neighbor(self, rank, nslave):