# Import firedrake stuff here so that adjoint mode works
from __future__ import absolute_import

import thetis_config

from firedrake import *

if thetis_config.adjoint:
    from firedrake_adjoint import *

@@ -135,16 +135,24 @@
 import *
+  # NOQA
 %0A%0Aif the
@@ -206,9 +206,17 @@
 import *
+  # NOQA
 %0A

Fix lint issues