Vendor rechunker.algorithm inside xarray-beam

Rechunker is deprecated and no longer maintained:
pangeo-data/rechunker#160

In particular, it pins zarr=2, but we need to support zarr v3 in xarray-beam.

PiperOrigin-RevId: 804545298

Author: shoyer

LICENSE

@@ -200,3 +200,25 @@
    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    See the License for the specific language governing permissions and
    limitations under the License.
+
+------------------
+
+Copyright (c) 2020 Rechunker Developers
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

pyproject.toml

@@ -20,7 +20,6 @@ dependencies = [
     "apache_beam>=2.31.0",
     "dask",
     "immutabledict",
-    "rechunker>=0.5.1",
     "zarr",
     "xarray",
 ]
@@ -32,10 +31,10 @@ documentation = "https://xarray-beam.readthedocs.io"
 [project.optional-dependencies]
 tests = [
     "absl-py",
+    "h5netcdf",
+    "hypothesis",
     "pandas",
     "pytest",
-    "scipy",
-    "h5netcdf",
 ]
 docs = [
     'myst-nb',

xarray_beam/__init__.py

@@ -52,4 +52,4 @@
     DatasetToZarr,
 )
 
-__version__ = '0.8.1'  # automatically synchronized to pyproject.toml
+__version__ = '0.9.0'  # automatically synchronized to pyproject.toml

xarray_beam/_src/rechunk.py

@@ -33,10 +33,10 @@
 
 import apache_beam as beam
 import numpy as np
-from rechunker import algorithm
 import xarray
 
 from xarray_beam._src import core
+from xarray_beam._src import rechunker_algorithm as algorithm
 
 
 # pylint: disable=logging-not-lazy