move most imports to the top

firedrakeproject · Feb 9, 2023 · e1380de · e1380de
1 parent ac722e7
commit e1380de
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/firedrake/preconditioners/hiptmair.py b/firedrake/preconditioners/hiptmair.py
@@ -2,6 +2,10 @@
 
 from firedrake.petsc import PETSc
 from firedrake.preconditioners.base import PCBase
+from firedrake.functionspace import FunctionSpace
+from firedrake.ufl_expr import TestFunction, TrialFunction
+from firedrake.preconditioners.hypre_ams import chop
+from firedrake.parameters import parameters
 from firedrake_citations import Citations
 from firedrake.interpolation import Interpolator
 from ufl.algorithms.ad import expand_derivatives
@@ -25,7 +29,6 @@ def coarsen(self, pc):
 
     def initialize(self, pc):
         from firedrake.assemble import allocate_matrix, TwoFormAssembler
-        from firedrake import parameters
         A, P = pc.getOperators()
         appctx = self.get_appctx(pc)
         fcp = appctx.get("form_compiler_parameters")
@@ -137,7 +140,6 @@ class HiptmairPC(TwoLevelPC):
     _prefix = "hiptmair_"
 
     def coarsen(self, pc):
-        from firedrake import FunctionSpace, TestFunction, TrialFunction
         Citations().register("Hiptmair1998")
         appctx = self.get_appctx(pc)
 
@@ -192,7 +194,6 @@ def coarsen(self, pc):
                 coarse_operator += beta(test, shift*trial, coefficients={})
 
         if G_callback is None:
-            from firedrake.preconditioners.hypre_ams import chop
             interp_petscmat = chop(Interpolator(dminus(test), V, bcs=bcs + coarse_space_bcs).callable().handle)
         else:
             interp_petscmat = G_callback(V, coarse_space, bcs, coarse_space_bcs)