Linting import orders

driver/examples/stencil_signatures.py

@@ -2,10 +2,10 @@
 import inspect
 from typing import Optional, TextIO
 
-import yaml
-
 import ndsl.dsl
 import ndsl.util
+import yaml
+
 import pace.driver
 
 

driver/pace/driver/configs/comm.py

@@ -4,7 +4,6 @@
 from typing import Any, ClassVar, List
 
 import dacite
-
 from ndsl.comm.caching_comm import CachingCommReader, CachingCommWriter
 from ndsl.comm.mpi import MPIComm
 from ndsl.comm.null_comm import NullComm

driver/pace/driver/diagnostics.py

@@ -11,6 +11,7 @@
 from ndsl.grid import GridData
 from ndsl.monitor import Monitor, NetCDFMonitor, ZarrMonitor
 from ndsl.quantity import Quantity
+
 from pace.fv3core.dycore_state import DycoreState
 
 from .state import DriverState

driver/pace/driver/driver.py

@@ -8,9 +8,6 @@
 import dace
 import dacite
 import yaml
-
-import pace.driver
-import pace.physics
 from ndsl.comm.comm_abc import Comm
 from ndsl.comm.communicator import (
     Communicator,
@@ -31,6 +28,9 @@
 from ndsl.performance import PerformanceConfig
 from ndsl.performance.collector import PerformanceCollector
 from ndsl.performance.timer import Timer
+
+import pace.driver
+import pace.physics
 from pace import fv3core
 from pace.driver.safety_checks import SafetyChecker
 

driver/pace/driver/grid.py

@@ -4,7 +4,6 @@
 
 import f90nml
 import xarray as xr
-
 from ndsl.comm.communicator import Communicator
 from ndsl.comm.partitioner import get_tile_index
 from ndsl.constants import X_DIM, X_INTERFACE_DIM, Y_DIM, Y_INTERFACE_DIM

driver/pace/driver/initialization.py

@@ -6,10 +6,6 @@
 from typing import Callable, ClassVar, List, Type, TypeVar
 
 import f90nml
-
-import pace.driver
-import pace.fv3core.initialization.analytic_init as analytic_init
-import pace.physics
 from ndsl.comm.communicator import Communicator
 from ndsl.constants import X_DIM, Y_DIM
 from ndsl.dsl.dace.orchestration import DaceConfig
@@ -19,6 +15,10 @@
 from ndsl.initialization.allocator import QuantityFactory
 from ndsl.namelist import Namelist
 from ndsl.stencils.testing import TranslateGrid, grid
+
+import pace.driver
+import pace.fv3core.initialization.analytic_init as analytic_init
+import pace.physics
 from pace import fv3core
 from pace.fv3core.testing import TranslateFVDynamics
 

driver/pace/driver/run.py

@@ -4,7 +4,6 @@
 
 import click
 import yaml
-
 from ndsl.logging import AVAILABLE_LOG_LEVELS, ndsl_log
 
 from .driver import Driver, DriverConfig

driver/pace/driver/safety_checks.py

@@ -1,8 +1,8 @@
 from typing import ClassVar, Dict, Optional
 
 import numpy as np
-
 from ndsl.quantity import Quantity
+
 from pace.fv3core.dycore_state import DycoreState
 
 

driver/pace/driver/state.py

@@ -2,10 +2,8 @@
 from dataclasses import fields
 from typing import List
 
-import xarray as xr
-
 import ndsl.dsl.gt4py_utils as gt_utils
-import pace.physics
+import xarray as xr
 from ndsl.comm.communicator import Communicator
 from ndsl.constants import N_HALO_DEFAULT, X_DIM, Y_DIM, Z_DIM
 from ndsl.dsl.typing import Float
@@ -14,6 +12,8 @@
 from ndsl.initialization.allocator import QuantityFactory
 from ndsl.initialization.sizer import SubtileGridSizer
 from ndsl.quantity import Quantity
+
+import pace.physics
 from pace import fv3core
 
 

driver/tests/mpi/test_restart.py

@@ -6,11 +6,11 @@
 import yaml
 import zarr
 from mpi4py import MPI
-
 from ndsl.comm.communicator import CubedSphereCommunicator
 from ndsl.comm.null_comm import NullComm
 from ndsl.comm.partitioner import CubedSpherePartitioner, TilePartitioner
 from ndsl.quantity import Quantity
+
 from pace.driver import DriverConfig
 from pace.driver.state import DriverState
 from pace.physics import PHYSICS_PACKAGES

examples/notebooks/functions.py

@@ -8,8 +8,6 @@
 from fv3viz import pcolormesh_cube
 from IPython.display import HTML, display
 from matplotlib import animation
-from units_config import units
-
 from ndsl.comm.communicator import CubedSphereCommunicator
 from ndsl.comm.partitioner import CubedSpherePartitioner, TilePartitioner
 from ndsl.constants import RADIUS
@@ -29,6 +27,8 @@
 from ndsl.initialization.allocator import QuantityFactory
 from ndsl.initialization.sizer import SubtileGridSizer
 from ndsl.quantity import Quantity
+from units_config import units
+
 from pace.fv3core.stencils.fvtp2d import FiniteVolumeTransport
 from pace.fv3core.stencils.fxadv import FiniteVolumeFluxPrep
 from pace.fv3core.stencils.tracer_2d_1l import TracerAdvection

fv3core/examples/standalone/runfile/acoustics.py

@@ -5,16 +5,16 @@
 
 import click
 import f90nml
-import serialbox
-import yaml
-from timing import collect_data_and_write_to_file
-
 import ndsl.dsl
 import ndsl.util as util
+import serialbox
+import yaml
 from ndsl.comm.null_comm import NullComm
 from ndsl.dsl.dace.orchestration import DaceConfig
 from ndsl.dsl.stencil import CompilationConfig
 from ndsl.stencils.testing.grid import Grid
+from timing import collect_data_and_write_to_file
+
 from pace.fv3core._config import DynamicalCoreConfig
 from pace.fv3core.stencils.dyn_core import AcousticDynamics
 from pace.fv3core.testing import TranslateDynCore

fv3core/examples/standalone/runfile/compile.py

@@ -7,9 +7,9 @@
 
 import f90nml
 import gt4py.cartesian.config
-
 import ndsl.dsl.stencil  # noqa: F401
 from ndsl.comm.null_comm import NullComm
+
 from pace.fv3core._config import DynamicalCoreConfig
 
 

fv3core/examples/standalone/runfile/dynamics.py

@@ -9,22 +9,23 @@
 from typing import Any, Dict, List, Tuple
 
 import f90nml
-import numpy as np
-import xarray as xr
-import yaml
-from mpi4py import MPI
 
 # NOTE: we need to import dsl.stencil prior to
 # ndsl.util, otherwise xarray precedes gt4py, causing
 # very strange errors on some systems (e.g. daint)
 import ndsl.dsl.stencil
 import ndsl.util as util
+import numpy as np
+import xarray as xr
+import yaml
+from mpi4py import MPI
 from ndsl.comm.null_comm import NullComm
 from ndsl.dsl import StencilFactory
 from ndsl.dsl.dace.orchestration import DaceConfig
 from ndsl.grid import DampingCoefficients, GridData, MetricTerms
 from ndsl.stencils.testing import dataset_to_dict
 from ndsl.stencils.testing.grid import Grid
+
 from pace.fv3core import DynamicalCore, DynamicalCoreConfig
 from pace.fv3core.dycore_state import DycoreState
 from pace.fv3core.initialization.baroclinic import init_baroclinic_state

fv3core/pace/fv3core/_config.py

@@ -2,7 +2,6 @@
 from typing import Optional, Tuple
 
 import f90nml
-
 from ndsl.namelist import Namelist, NamelistDefaults
 
 

fv3core/pace/fv3core/dycore_state.py

@@ -1,9 +1,8 @@
 from dataclasses import asdict, dataclass, field, fields
 from typing import Any, Dict, Mapping, Union
 
-import xarray as xr
-
 import ndsl.dsl.gt4py_utils as gt_utils
+import xarray as xr
 from ndsl.comm.communicator import Communicator
 from ndsl.constants import (
     X_DIM,

fv3core/pace/fv3core/initialization/analytic_init.py

@@ -4,6 +4,7 @@
 from ndsl.grid import GridData
 from ndsl.initialization.allocator import QuantityFactory
 from ndsl.utils import MetaEnumStr
+
 from pace.fv3core.dycore_state import DycoreState
 
 

fv3core/pace/fv3core/initialization/init_utils.py

@@ -2,12 +2,12 @@
 from dataclasses import fields
 from types import SimpleNamespace
 
-import numpy as np
-
 import ndsl.constants as constants
+import numpy as np
 from ndsl.dsl.typing import Float
 from ndsl.grid import lon_lat_midpoint
 from ndsl.grid.gnomonic import get_lonlat_vect, get_unit_vector_direction
+
 from pace.fv3core.dycore_state import DycoreState
 
 

fv3core/pace/fv3core/initialization/test_cases/initialize_baroclinic.py

@@ -1,13 +1,13 @@
 import math
 
-import numpy as np
-
 import ndsl.constants as constants
 import ndsl.dsl.gt4py_utils as utils
-import pace.fv3core.initialization.init_utils as init_utils
+import numpy as np
 from ndsl.comm.communicator import CubedSphereCommunicator
 from ndsl.grid import GridData, great_circle_distance_lon_lat, lon_lat_midpoint
 from ndsl.initialization.allocator import QuantityFactory
+
+import pace.fv3core.initialization.init_utils as init_utils
 from pace.fv3core.dycore_state import DycoreState
 
 

fv3core/pace/fv3core/initialization/test_cases/initialize_tc.py

@@ -1,10 +1,10 @@
-import numpy as np
-
 import ndsl.constants as constants
-import pace.fv3core.initialization.init_utils as init_utils
+import numpy as np
 from ndsl.comm.communicator import CubedSphereCommunicator
 from ndsl.grid import GridData, great_circle_distance_lon_lat
 from ndsl.initialization.allocator import QuantityFactory
+
+import pace.fv3core.initialization.init_utils as init_utils
 from pace.fv3core.dycore_state import DycoreState
 
 

fv3core/pace/fv3core/stencils/a2b_ord4.py

@@ -10,13 +10,13 @@
     sin,
     sqrt,
 )
-
 from ndsl.constants import X_DIM, X_INTERFACE_DIM, Y_DIM, Y_INTERFACE_DIM, Z_DIM
 from ndsl.dsl.dace.orchestration import orchestrate
 from ndsl.dsl.stencil import GridIndexing, StencilFactory
 from ndsl.dsl.typing import Float, FloatField, FloatFieldI, FloatFieldIJ
 from ndsl.grid import GridData
 from ndsl.initialization.allocator import QuantityFactory
+
 from pace.fv3core.stencils.basic_operations import copy_defn
 
 

fv3core/pace/fv3core/stencils/basic_operations.py

@@ -1,6 +1,5 @@
 import gt4py.cartesian.gtscript as gtscript
 from gt4py.cartesian.gtscript import PARALLEL, computation, interval
-
 from ndsl.dsl.typing import Float, FloatField, FloatFieldIJ
 
 

fv3core/pace/fv3core/stencils/c_sw.py

@@ -6,7 +6,6 @@
     interval,
     region,
 )
-
 from ndsl.constants import X_DIM, X_INTERFACE_DIM, Y_DIM, Y_INTERFACE_DIM, Z_DIM
 from ndsl.dsl.dace.orchestration import orchestrate
 from ndsl.dsl.stencil import StencilFactory
@@ -15,6 +14,7 @@
 from ndsl.initialization.allocator import QuantityFactory
 from ndsl.quantity import Quantity
 from ndsl.stencils import corners
+
 from pace.fv3core.stencils.d2a2c_vect import DGrid2AGrid2CGridVectors
 
 

fv3core/pace/fv3core/stencils/d2a2c_vect.py

@@ -1,13 +1,13 @@
 import gt4py.cartesian.gtscript as gtscript
 from gt4py.cartesian.gtscript import PARALLEL, computation, horizontal, interval, region
-
 from ndsl.constants import X_DIM, Y_DIM, Z_DIM
 from ndsl.dsl.dace.orchestration import orchestrate
 from ndsl.dsl.stencil import StencilFactory
 from ndsl.dsl.typing import Float, FloatField, FloatFieldIJ
 from ndsl.grid import GridData
 from ndsl.initialization.allocator import QuantityFactory
 from ndsl.stencils import corners
+
 from pace.fv3core.stencils.a2b_ord4 import a1, a2, lagrange_x_func, lagrange_y_func
 
 

fv3core/pace/fv3core/stencils/d_sw.py

@@ -9,15 +9,15 @@
     interval,
     region,
 )
-
-import pace.fv3core.stencils.delnflux as delnflux
 from ndsl.constants import X_DIM, X_INTERFACE_DIM, Y_DIM, Y_INTERFACE_DIM, Z_DIM
 from ndsl.dsl.dace.orchestration import orchestrate
 from ndsl.dsl.stencil import StencilFactory
 from ndsl.dsl.typing import Float, FloatField, FloatFieldIJ, FloatFieldK
 from ndsl.grid import DampingCoefficients, GridData
 from ndsl.initialization.allocator import QuantityFactory
 from ndsl.quantity import Quantity
+
+import pace.fv3core.stencils.delnflux as delnflux
 from pace.fv3core._config import DGridShallowWaterLagrangianDynamicsConfig
 from pace.fv3core.stencils.d2a2c_vect import contravariant
 from pace.fv3core.stencils.delnflux import DelnFluxNoSG

fv3core/pace/fv3core/stencils/del2cubed.py

@@ -1,12 +1,12 @@
-from gt4py.cartesian.gtscript import PARALLEL, computation, horizontal, interval, region
-
 import ndsl.stencils.corners as corners
+from gt4py.cartesian.gtscript import PARALLEL, computation, horizontal, interval, region
 from ndsl.constants import X_DIM, X_INTERFACE_DIM, Y_DIM, Y_INTERFACE_DIM, Z_DIM
 from ndsl.dsl.dace.orchestration import orchestrate
 from ndsl.dsl.stencil import StencilFactory, get_stencils_with_varied_bounds
 from ndsl.dsl.typing import Float, FloatField, FloatFieldIJ, cast_to_index3d
 from ndsl.grid import DampingCoefficients
 from ndsl.initialization.allocator import QuantityFactory
+
 from pace.fv3core.stencils.basic_operations import copy_defn
 
 

fv3core/pace/fv3core/stencils/delnflux.py

@@ -9,7 +9,6 @@
     interval,
     region,
 )
-
 from ndsl.constants import X_DIM, X_INTERFACE_DIM, Y_DIM, Y_INTERFACE_DIM, Z_DIM
 from ndsl.dsl.dace.orchestration import orchestrate
 from ndsl.dsl.stencil import StencilFactory, get_stencils_with_varied_bounds