misc: Isort and flake8 petsc branch since merge with main

Author: ZoeLeibowitz

.github/workflows/pytest-core-mpi.yaml

@@ -12,9 +12,11 @@ on:
   push:
     branches:
     - main
+    - petsc
   pull_request:
     branches:
     - main
+    - petsc
 
 jobs:
   test-mpi-basic:

.github/workflows/tutorials.yaml

@@ -12,9 +12,11 @@ on:
   push:
     branches:
       - main
+      - petsc
   pull_request:
     branches:
       - main
+      - petsc
 
 jobs:
   tutorials-basic:

conftest.py

@@ -17,8 +17,8 @@
 from devito.ir.iet import (
     FindNodes, FindSymbols, Iteration, ParallelBlock, retrieve_iteration_tree
 )
-from devito.tools import as_tuple
 from devito.petsc.config import PetscOSError, get_petsc_dir
+from devito.tools import as_tuple
 
 try:
     from mpi4py import MPI  # noqa

devito/core/cpu.py

@@ -10,12 +10,11 @@
 )
 from devito.passes.equations import collect_derivatives
 from devito.passes.iet import (
-    COmpTarget, CTarget, CXXOmpTarget, CXXTarget, avoid_denormals, check_stability,
-    hoist_prodders, linearize, mpiize, relax_incr_dimensions, PetscTarget
+    COmpTarget, CTarget, CXXOmpTarget, CXXTarget, PetscTarget, avoid_denormals,
+    check_stability, hoist_prodders, linearize, mpiize, relax_incr_dimensions
 )
-from devito.tools import timed_pass
-
 from devito.petsc.iet.passes import lower_petsc_symbols
+from devito.tools import timed_pass
 
 __all__ = [
     'Cpu64AdvCOperator',

devito/ir/iet/algorithms.py

@@ -5,9 +5,9 @@
     Section, Switch, SyncSpot
 )
 from devito.ir.support import GuardCaseSwitch, GuardSwitch
-from devito.tools import as_mapper, timed_pass
-from devito.petsc.types import MetaData
 from devito.petsc.iet.nodes import petsc_iet_mapper
+from devito.petsc.types import MetaData
+from devito.tools import as_mapper, timed_pass
 
 __all__ = ['iet_build']
 

devito/ir/iet/nodes.py

@@ -23,7 +23,7 @@
     Signer, as_tuple, ctypes_to_cstr, filter_ordered, filter_sorted, flatten
 )
 from devito.types.basic import AbstractFunction, AbstractSymbol, Basic, Indexed, Symbol
-from devito.types.object import AbstractObject, LocalObject, LocalCompositeObject
+from devito.types.object import AbstractObject, LocalCompositeObject, LocalObject
 
 __all__ = [
     'AugmentedExpression',

devito/ir/iet/visitors.py

@@ -29,8 +29,8 @@
     is_external_ctype, sorted_priority
 )
 from devito.types import (
-    ArrayObject, CompositeObject, DeviceMap, Dimension, IndexedData, Pointer,
-    LocalCompositeObject
+    ArrayObject, CompositeObject, DeviceMap, Dimension, IndexedData, LocalCompositeObject,
+    Pointer
 )
 from devito.types.basic import AbstractFunction, AbstractSymbol, Basic
 

devito/operator/operator.py

@@ -35,6 +35,9 @@
     Graph, error_mapper, generate_implicit, generate_macros, is_on_device, lower_dtypes,
     lower_index_derivatives, minimize_symbols, optimize_pows, unevaluate
 )
+from devito.petsc.clusters import petsc_preprocess
+from devito.petsc.equations import lower_exprs_petsc
+from devito.petsc.iet.passes import lower_petsc
 from devito.symbolics import estimate_cost, subs_op_args
 from devito.tools import (
     DAG, CacheInstances, MemoryEstimate, OrderedSet, ReducerMap, Signer, as_mapper,
@@ -43,9 +46,6 @@
 )
 from devito.types import Buffer, Evaluable, device_layer, disk_layer, host_layer
 from devito.types.dimension import Thickness
-from devito.petsc.iet.passes import lower_petsc
-from devito.petsc.clusters import petsc_preprocess
-from devito.petsc.equations import lower_exprs_petsc
 from devito.warnings import warn
 
 __all__ = ['Operator']

devito/operator/profiling.py

@@ -16,9 +16,9 @@
 from devito.logger import warning
 from devito.mpi import MPI
 from devito.parameters import configuration
+from devito.petsc.logging import PetscSummary
 from devito.symbolics import subs_op_args
 from devito.tools import DefaultOrderedDict, flatten
-from devito.petsc.logging import PetscSummary
 
 __all__ = ['create_profile']
 

devito/passes/iet/engine.py

@@ -23,9 +23,9 @@
     Temp, ThreadArray, Wildcard
 )
 from devito.types.args import ArgProvider
+from devito.types.array import ArrayBasic
 from devito.types.dense import DiscreteFunction
 from devito.types.dimension import AbstractIncrDimension, BlockDimension
-from devito.types.array import ArrayBasic
 
 __all__ = ['Graph', 'iet_pass', 'iet_visit']