misc: Clean up internals

Author: ZoeLeibowitz

devito/petsc/iet/callback_builder.py

@@ -1,8 +1,9 @@
 from collections import OrderedDict
 
-from devito.ir.iet import (Call, FindSymbols, List, Uxreplace, CallableBody,
-                           Dereference, DummyExpr, BlankLine, Callable, Iteration,
-                           PointerCast, Definition)
+from devito.ir.iet import (
+    Call, FindSymbols, List, Uxreplace, CallableBody, Dereference, DummyExpr,
+    BlankLine, Callable, Iteration, PointerCast, Definition
+)
 from devito.symbolics import (
     Byref, FieldFromPointer, IntDiv, Deref, Mod, String, Null, VOID
 )

devito/petsc/iet/object_builder.py

@@ -4,12 +4,12 @@
 from devito.types import Symbol
 from devito.tools import frozendict
 
-from devito.petsc.types import (PetscBundle, DM, Mat, CallbackVec, Vec,
-                                KSP, PC, SNES, PetscInt, StartPtr, PointerIS, PointerDM,
-                                VecScatter, JacobianStruct, SubMatrixStruct,
-                                CallbackDM, PetscMPIInt, PetscErrorCode, PointerMat,
-                                MatReuse, CallbackPointerDM, CallbackPointerIS,
-                                CallbackMat, DummyArg, NofSubMats)
+from devito.petsc.types import (
+    PetscBundle, DM, Mat, CallbackVec, Vec, KSP, PC, SNES, PetscInt, StartPtr,
+    PointerIS, PointerDM, VecScatter, JacobianStruct, SubMatrixStruct, CallbackDM,
+    PetscMPIInt, PetscErrorCode, PointerMat, MatReuse, CallbackPointerDM,
+    CallbackPointerIS, CallbackMat, DummyArg, NofSubMats
+)
 
 
 class BaseObjectBuilder:

devito/petsc/iet/run_solver.py

@@ -3,7 +3,7 @@
 from devito.ir.iet import (
     BlankLine, FindNodes, retrieve_iteration_tree, filter_iterations
 )
-from devito.symbolics import (Byref, Null)
+from devito.symbolics import Byref, Null
 
 from devito.petsc.iet.nodes import PetscMetaData, petsc_call
 from devito.petsc.types.modes import InsertMode, ScatterMode

devito/petsc/initialize.py

@@ -6,6 +6,7 @@
 from devito import Operator, switchconfig
 from devito.types import Symbol
 from devito.types.equation import PetscEq
+
 from devito.petsc.types import Initialize, Finalize
 
 global _petsc_initialized

devito/petsc/solve.py

@@ -4,12 +4,14 @@
 from devito.operations.solve import eval_time_derivatives
 from devito.symbolics import retrieve_functions, retrieve_dimensions
 
-from devito.petsc.types import (LinearSolverMetaData, PETScArray, DMDALocalInfo,
-                                FieldData, MultipleFieldData, Jacobian, Residual,
-                                MixedResidual, MixedJacobian, InitialGuess)
+from devito.petsc.types import (
+    LinearSolverMetaData, PETScArray, DMDALocalInfo, FieldData, MultipleFieldData,
+    Jacobian, Residual, MixedResidual, MixedJacobian, InitialGuess
+)
 from devito.petsc.types.equation import EssentialBC
-from devito.petsc.solver_parameters import (linear_solver_parameters,
-                                            format_options_prefix)
+from devito.petsc.solver_parameters import (
+    linear_solver_parameters, format_options_prefix
+)
 
 
 __all__ = ['PETScSolve']

devito/petsc/solver_parameters.py

@@ -1,6 +1,7 @@
-from petsctools import flatten_parameters
 import itertools
 
+from petsctools import flatten_parameters
+
 
 # NOTE: Will be extended, the default preconditioner is not going to be 'none'
 base_solve_defaults = {

devito/petsc/types/modes.py

@@ -1,4 +1,3 @@
-
 class InsertMode:
     """
     How the entries are combined with the current values in the vectors or matrices.

devito/petsc/types/object.py

@@ -1,8 +1,10 @@
 from ctypes import POINTER, c_char
 
 from devito.tools import CustomDtype, dtype_to_ctype, as_tuple, CustomIntType
-from devito.types import (LocalObject, LocalCompositeObject, ModuloDimension,
-                          TimeDimension, ArrayObject, CustomDimension, Scalar)
+from devito.types import (
+    LocalObject, LocalCompositeObject, ModuloDimension, TimeDimension, ArrayObject,
+    CustomDimension, Scalar
+)
 from devito.symbolics import Byref, cast
 from devito.types.basic import DataSymbol, LocalType