Cleanup refresh to make the cleanup optional and avoid taking the lock

theano/gof/compilelock.py

@@ -7,6 +7,7 @@ import random
 import socket  # only used for gethostname()
 import time
 import logging
+import contextlib
 
 from theano import config
 from theano.configparser import AddConfigVar, IntParam
@@ -44,6 +45,14 @@ def force_unlock():
     release_lock()
 
 
+@contextmanager
+def lock_ctx(lock_dir=None, keep_lock=False, **kw):
+    get_lock(lock_dir=lock_dir, **kw)
+    yield
+    if not keep_lock:
+        release_lock()
+
+
 def get_lock(lock_dir=None, **kw):
     """
     Obtain lock on compilation directory.