more pylint cleanup

jupe · Oct 2, 2020 · 7c63fe4 · 7c63fe4
1 parent 3ff704f
commit 7c63fe4
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/lockable/plugin.py b/lockable/plugin.py
@@ -70,8 +70,8 @@ def release():
             except OSError as error:
                 print(error, file=sys.stderr)
         return candidate, release
-    except Timeout:
-        raise AssertionError('not success')
+    except Timeout as error:
+        raise AssertionError('not success') from error
 
 
 @contextmanager
@@ -93,8 +93,8 @@ def doit(candidates_inner):
         print(f'resource {resource["id"]} allocated ({json.dumps(resource)})')
         yield resource
         release()
-    except FunctionTimedOut:
-        raise TimeoutError(f'Allocation timeout ({timeout_s}s)')
+    except FunctionTimedOut as error:
+        raise TimeoutError(f'Allocation timeout ({timeout_s}s)') from error
 
 
 @contextmanager