Detecting if we are inside a job

quake/job/__main__.py

@@ -5,6 +5,7 @@ import logging
 import uvloop
 
 from .job import Job
+from .globals import set_job_flag
 
 uvloop.install()
 
@@ -18,6 +19,7 @@ def parse_args():
 
 
 async def main():
+    set_job_flag()
     args = parse_args()
     job = Job(args.task_id, args.rank, args.ds_local_port, {})
     await job.start()

quake/job/globals.py

+
+_is_inside_job = False
+
+
+def is_inside_job():
+    return _is_inside_job
+
+
+def set_job_flag():
+    global _is_inside_job
+    _is_inside_job = True