Reformat Xee codebase using pyink

conftest.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 # ==============================================================================
 """Configure FLAGS with default values for absltest."""
+
 from absl import app
 
 try:

docs/conf.py

@@ -24,10 +24,10 @@
 print('python exec:', sys.executable)
 print('sys.path:', sys.path)
 if importlib.util.find_spec('pip') is not None:
-    print('pip environment:')
-    subprocess.run([sys.executable, '-m', 'pip', 'list'])  # pylint: disable=subprocess-run-check
+  print('pip environment:')
+  subprocess.run([sys.executable, '-m', 'pip', 'list'])  # pylint: disable=subprocess-run-check
 else:
-    print('pip environment: unavailable')
+  print('pip environment: unavailable')
 
 print(f'xee: {xee.__file__}')
 

examples/dataflow/ee_to_zarr_dataflow.py

@@ -14,17 +14,16 @@
 # ==============================================================================
 r"""Exports EE ImageCollections to Zarr using Xarray-Beam."""
 
-
 import logging
 
 from absl import app
 from absl import flags
 import apache_beam as beam
+import shapely
 import xarray as xr
 import xarray_beam as xbeam
 import xee
 from xee import helpers
-import shapely
 
 import ee
 

examples/ee_to_zarr.py

@@ -19,15 +19,14 @@
 from absl import app
 from absl import flags
 import apache_beam as beam
+import shapely
 import xarray as xr
 import xarray_beam as xbeam
 import xee
 from xee import helpers
-import shapely
 
 import ee
 
-
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.INFO)
 

xee/__init__.py

@@ -27,7 +27,10 @@
 
 from .ext import *  # noqa: F401,F403  (backend classes)
 from .ext import __version__  # noqa: F401
-from .helpers import fit_geometry, extract_grid_params, set_scale, PixelGridParams  # noqa: F401
+from .helpers import extract_grid_params  # noqa: F401
+from .helpers import fit_geometry  # noqa: F401
+from .helpers import PixelGridParams  # noqa: F401
+from .helpers import set_scale  # noqa: F401
 
 __all__ = [
     # Version.

xee/ext.py

@@ -45,7 +45,6 @@
 
 import ee
 
-
 assert sys.version_info >= (3, 10)
 try:
   __version__ = importlib.metadata.version('xee') or 'unknown'
@@ -956,9 +955,8 @@ def open_dataset(
     Args:
       filename_or_obj: An asset ID for an ImageCollection, or an
         ``ee.ImageCollection`` object.
-      crs: The coordinate reference system (a CRS code or WKT
-        string). This defines the frame of reference to coalesce all variables
-        upon opening.
+      crs: The coordinate reference system (a CRS code or WKT string). This
+        defines the frame of reference to coalesce all variables upon opening.
       crs_transform: Transform matrix describing the grid origin and scale
         relative to the CRS.
       shape_2d: Dimensions of the pixel grid in the form (width, height).
@@ -977,8 +975,8 @@ def open_dataset(
       decode_timedelta (optional): If True, decode variables and coordinates
         with time units in {"days", "hours", "minutes", "seconds",
         "milliseconds", "microseconds"} into timedelta objects. If False, leave
-        them encoded as numbers. If ``None`` (default), assume the same value
-        of ``decode_times``.
+        them encoded as numbers. If ``None`` (default), assume the same value of
+        ``decode_times``.
       use_cftime (optional): Only relevant if encoded dates come from a standard
         calendar (e.g. "gregorian", "proleptic_gregorian", "standard", or not
         specified).  If None (default), attempt to decode times to
@@ -991,12 +989,12 @@ def open_dataset(
       concat_characters (optional): Should character arrays be concatenated to
         strings, for example: ``["h", "e", "l", "l", "o"] -> "hello"``.
       decode_coords (optional): ``bool`` or ``{"coordinates", "all"}``. Controls
-        which variables are set as coordinate variables. Use
-        ``"coordinates"`` (or ``True``) to set variables referenced by the
-        ``'coordinates'`` attribute of datasets or individual variables as
-        coordinate variables. Use ``"all"`` to additionally set variables
-        referenced by ``'grid_mapping'``, ``'bounds'``, and related attributes
-        as coordinate variables.
+        which variables are set as coordinate variables. Use ``"coordinates"``
+        (or ``True``) to set variables referenced by the ``'coordinates'``
+        attribute of datasets or individual variables as coordinate variables.
+        Use ``"all"`` to additionally set variables referenced by
+        ``'grid_mapping'``, ``'bounds'``, and related attributes as coordinate
+        variables.
       primary_dim_name (optional): Override the name of the primary dimension of
         the output Dataset. By default, the name is ``'time'``.
       primary_dim_property (optional): Override the ``ee.Image`` property for
@@ -1023,6 +1021,7 @@ def open_dataset(
         makes slicing an ImageCollection across time faster. This optimization
         loads EE images in a slice by ID, so any modifications to images in a
         computed ImageCollection will not be reflected.
+
     Returns:
       An xarray.Dataset that streams in remote data from Earth Engine.
     """

xee/ext_integration_test.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 # ==============================================================================
 r"""Integration tests for the Google Earth Engine backend for Xarray."""
+
 import json
 import os
 import pathlib

xee/ext_test.py

@@ -1,11 +1,11 @@
 """Xee Unit Tests."""
 
+from unittest import mock
 from absl.testing import absltest
 from absl.testing import parameterized
-import numpy as np
 import affine
+import numpy as np
 import shapely
-from unittest import mock
 import xee
 from xee import ext
 from xee import helpers

xee/micro_benchmarks.py

@@ -30,7 +30,6 @@
 
 import ee
 
-
 REPEAT = 10
 LOOPS = 1
 PROFILE = False

xee/types.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 # ==============================================================================
 """Type definitions for Earth Engine concepts (and others)."""
+
 from typing import TypedDict, Union
 
 TileIndex = tuple[int, int, int]