doc string

dmlc · Jan 5, 2019 · c378481 · c378481
1 parent 2163332
commit c378481
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/dgl/runtime/ir/program.py b/python/dgl/runtime/ir/program.py
@@ -46,15 +46,17 @@ def pprint(self):
             self.pprint_exe(exe)
 
 class CurrentProgram(threading.local):
-
+    """Thread local storage to keep the reference of current thread's program"""
     def __init__(self):
         super(CurrentProgram, self).__init__()
         self.prog = None
 
     def get_prog(self):
+        """Get program"""
         return self.prog
 
     def set_prog(self, program):
+        """Set program"""
         self.prog = program
 
 # current program