diff --git a/pyiceberg/table/theta_sketch.py b/pyiceberg/table/theta_sketch.py new file mode 100644 index 0000000000..aa6c121094 --- /dev/null +++ b/pyiceberg/table/theta_sketch.py @@ -0,0 +1,74 @@ +# Licensed to the Apache Software Foundation (ASF) under one +# or more contributor license agreements. See the NOTICE file +# distributed with this work for additional information +# regarding copyright ownership. The ASF licenses this file +# to you under the Apache License, Version 2.0 (the +# "License"); you may not use this file except in compliance +# with the License. You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, +# software distributed under the License is distributed on an +# "AS IS" BASIS, 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. +from __future__ import annotations + +from typing import TYPE_CHECKING + +import zstandard + +from pyiceberg.table.puffin import PuffinBlobMetadata, PuffinFile + +if TYPE_CHECKING: + from datasketches import compact_theta_sketch + +BLOB_TYPE_APACHE_DATASKETCHES_THETA_V1 = "apache-datasketches-theta-v1" + + +class ThetaSketch: + field_id: int + _sketch: compact_theta_sketch + + def __init__(self, field_id: int, sketch: compact_theta_sketch) -> None: + self.field_id = field_id + self._sketch = sketch + + def get_estimate(self) -> float: + return self._sketch.get_estimate() + + def get_lower_bound(self, num_std_devs: int = 1) -> float: + return self._sketch.get_lower_bound(num_std_devs) + + def get_upper_bound(self, num_std_devs: int = 1) -> float: + return self._sketch.get_upper_bound(num_std_devs) + + def is_empty(self) -> bool: + return self._sketch.is_empty() + + def is_estimation_mode(self) -> bool: + return self._sketch.is_estimation_mode() + + @property + def sketch(self) -> compact_theta_sketch: + return self._sketch + + +def _theta_sketches_from_blob(blob: PuffinBlobMetadata, payload: bytes) -> list[ThetaSketch]: + from datasketches import compact_theta_sketch + + if blob.compression_codec == "zstd": + payload = zstandard.decompress(payload) + + sketch = compact_theta_sketch.deserialize(payload) + return [ThetaSketch(field_id=field_id, sketch=sketch) for field_id in blob.fields] + + +def theta_sketches_from_puffin_file(puffin_file: PuffinFile) -> list[ThetaSketch]: + sketches = [] + for blob in puffin_file.footer.blobs: + if blob.type == BLOB_TYPE_APACHE_DATASKETCHES_THETA_V1: + sketches.extend(_theta_sketches_from_blob(blob, puffin_file.get_blob_payload(blob))) + return sketches diff --git a/pyproject.toml b/pyproject.toml index fdcac80ade..9097e26706 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -98,6 +98,7 @@ datafusion = ["datafusion>=52,<53"] gcp-auth = ["google-auth>=2.4.0"] entra-auth = ["azure-identity>=1.25.1"] geoarrow = ["geoarrow-pyarrow>=0.2.0"] +datasketches = ["datasketches>=3.4.0,<6.0.0"] [dependency-groups] dev = [ @@ -124,6 +125,7 @@ dev = [ "papermill>=2.6.0", "nbformat>=5.10.0", "ipykernel>=6.29.0", + "datasketches>=3.4.0,<6.0.0", ] # for mkdocs docs = [ diff --git a/tests/table/puffin/v1/theta-sketches.puffin b/tests/table/puffin/v1/theta-sketches.puffin new file mode 100644 index 0000000000..9beca220d0 Binary files /dev/null and b/tests/table/puffin/v1/theta-sketches.puffin differ diff --git a/tests/table/test_theta_sketch.py b/tests/table/test_theta_sketch.py new file mode 100644 index 0000000000..e1ad0e88d3 --- /dev/null +++ b/tests/table/test_theta_sketch.py @@ -0,0 +1,166 @@ +# Licensed to the Apache Software Foundation (ASF) under one +# or more contributor license agreements. See the NOTICE file +# distributed with this work for additional information +# regarding copyright ownership. The ASF licenses this file +# to you under the Apache License, Version 2.0 (the +# "License"); you may not use this file except in compliance +# with the License. You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, +# software distributed under the License is distributed on an +# "AS IS" BASIS, 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. +import json +from os import path + +import pytest +from datasketches import compact_theta_sketch, update_theta_sketch + +from pyiceberg.table.puffin import MAGIC_BYTES, PuffinFile +from pyiceberg.table.theta_sketch import ThetaSketch, theta_sketches_from_puffin_file + + +def _open_fixture(file: str) -> bytes: + cur_dir = path.dirname(path.realpath(__file__)) + with open(f"{cur_dir}/puffin/v1/{file}", "rb") as f: + return f.read() + + +def _make_sketch(values: list[int]) -> compact_theta_sketch: + ts = update_theta_sketch() + for v in values: + ts.update(v) + return ts.compact() + + +@pytest.fixture +def empty_sketch_bytes() -> bytes: + return update_theta_sketch().compact().serialize() + + +@pytest.fixture +def three_value_sketch_bytes() -> bytes: + return _make_sketch([1, 2, 3]).serialize() + + +def test_empty_sketch(empty_sketch_bytes: bytes) -> None: + sketch = compact_theta_sketch.deserialize(empty_sketch_bytes) + ts = ThetaSketch(field_id=1, sketch=sketch) + + assert ts.is_empty() + assert ts.get_estimate() == 0.0 + + +def test_sketch_estimate(three_value_sketch_bytes: bytes) -> None: + sketch = compact_theta_sketch.deserialize(three_value_sketch_bytes) + ts = ThetaSketch(field_id=1, sketch=sketch) + + assert not ts.is_empty() + assert ts.get_estimate() == pytest.approx(3.0) + assert not ts.is_estimation_mode() + + +def test_sketch_bounds_exact_mode(three_value_sketch_bytes: bytes) -> None: + sketch = compact_theta_sketch.deserialize(three_value_sketch_bytes) + ts = ThetaSketch(field_id=1, sketch=sketch) + + assert ts.get_lower_bound(1) == pytest.approx(3.0) + assert ts.get_upper_bound(1) == pytest.approx(3.0) + + +def test_sketch_field_id() -> None: + sketch = _make_sketch([10, 20, 30]) + ts = ThetaSketch(field_id=42, sketch=sketch) + + assert ts.field_id == 42 + + +def test_sketch_property() -> None: + sketch = _make_sketch([1, 2]) + ts = ThetaSketch(field_id=1, sketch=sketch) + + assert ts.sketch is sketch + + +def test_estimation_mode() -> None: + ts_builder = update_theta_sketch(lg_k=5) + for i in range(100): + ts_builder.update(i) + sketch = ts_builder.compact() + ts = ThetaSketch(field_id=1, sketch=sketch) + + assert ts.is_estimation_mode() + assert ts.get_estimate() > 0 + assert ts.get_lower_bound(1) <= ts.get_estimate() + assert ts.get_upper_bound(1) >= ts.get_estimate() + + +def _build_puffin_file(blob_bytes: bytes, field_ids: list[int], snapshot_id: int = 1) -> bytes: + # Puffin layout: magic(4) + blobs + footer_json + footer_size(4) + flags(4) + magic(4) + # Blob offsets are file-absolute; first blob starts immediately after the 4-byte magic. + blob_offset = 4 + footer = { + "blobs": [ + { + "type": "apache-datasketches-theta-v1", + "snapshot-id": snapshot_id, + "sequence-number": 1, + "fields": field_ids, + "offset": blob_offset, + "length": len(blob_bytes), + } + ], + "properties": {}, + } + footer_json = json.dumps(footer, separators=(",", ":")).encode("utf-8") + footer_size_bytes = len(footer_json).to_bytes(4, byteorder="little") + flags = b"\x00\x00\x00\x00" + return MAGIC_BYTES + blob_bytes + footer_json + footer_size_bytes + flags + MAGIC_BYTES + + +def test_theta_sketches_from_puffin_file_single_field(three_value_sketch_bytes: bytes) -> None: + puffin_bytes = _build_puffin_file(three_value_sketch_bytes, field_ids=[5]) + puffin_file = PuffinFile(puffin_bytes) + + sketches = theta_sketches_from_puffin_file(puffin_file) + + assert len(sketches) == 1 + assert sketches[0].field_id == 5 + assert sketches[0].get_estimate() == pytest.approx(3.0) + + +def test_theta_sketches_from_puffin_file_multiple_fields(three_value_sketch_bytes: bytes) -> None: + puffin_bytes = _build_puffin_file(three_value_sketch_bytes, field_ids=[1, 2, 3]) + puffin_file = PuffinFile(puffin_bytes) + + sketches = theta_sketches_from_puffin_file(puffin_file) + + assert len(sketches) == 3 + assert [s.field_id for s in sketches] == [1, 2, 3] + for sketch in sketches: + assert sketch.get_estimate() == pytest.approx(3.0) + + +def test_theta_sketches_from_puffin_file_empty_sketch(empty_sketch_bytes: bytes) -> None: + puffin_bytes = _build_puffin_file(empty_sketch_bytes, field_ids=[7]) + puffin_file = PuffinFile(puffin_bytes) + + sketches = theta_sketches_from_puffin_file(puffin_file) + + assert len(sketches) == 1 + assert sketches[0].is_empty() + assert sketches[0].get_estimate() == 0.0 + + +def test_theta_sketches_from_trino_written_puffin_file() -> None: + puffin_file = PuffinFile(_open_fixture("theta-sketches.puffin")) + sketches = theta_sketches_from_puffin_file(puffin_file) + + assert len(sketches) == 3 + assert [s.field_id for s in sketches] == [1, 2, 3] + for sketch in sketches: + assert sketch.get_estimate() == pytest.approx(5.0) diff --git a/uv.lock b/uv.lock index 896e7f42f5..695ae57eb0 100644 --- a/uv.lock +++ b/uv.lock @@ -1267,6 +1267,45 @@ wheels = [ { url = "https://files.pythonhosted.org/packages/07/6c/aa3f2f849e01cb6a001cd8554a88d4c77c5c1a31c95bdf1cf9301e6d9ef4/defusedxml-0.7.1-py2.py3-none-any.whl", hash = "sha256:a352e7e428770286cc899e2542b6cdaedb2b4953ff269a210103ec58f6198a61", size = 25604, upload-time = "2021-03-08T10:59:24.45Z" }, ] +[[package]] +name = "datasketches" +version = "5.2.0" +source = { registry = "https://pypi.org/simple" } +dependencies = [ + { name = "numpy" }, +] +sdist = { url = "https://files.pythonhosted.org/packages/34/cd/659ae9fc53f34d6deafbe12162977654be5bb0a584e6afa6656337e13952/datasketches-5.2.0.tar.gz", hash = "sha256:c00d61da4695e00036e63f590999f584cc39246cbb147b171f375f792604a612", size = 53213, upload-time = "2025-03-01T07:49:24.567Z" } +wheels = [ + { url = "https://files.pythonhosted.org/packages/9d/13/0146b96195819f528ecf3f77e0dd58054c321076468746fc67687c20ff19/datasketches-5.2.0-cp310-cp310-macosx_10_14_x86_64.whl", hash = "sha256:177e9545aafd7359c44e7935b995b0b0f9b08704a78f53e7840a7125d2d1fc9e", size = 652707, upload-time = "2025-03-01T07:48:01.904Z" }, + { url = "https://files.pythonhosted.org/packages/38/16/38fc321557d86be3542e63ab0c54eef4807c43a16e4d102b85fab64716d0/datasketches-5.2.0-cp310-cp310-macosx_11_0_arm64.whl", hash = "sha256:4d33a1d7894509556efb6a41f2ad530257d67a094a039e3b5a047a22934a951e", size = 584959, upload-time = "2025-03-01T07:48:03.875Z" }, + { url = "https://files.pythonhosted.org/packages/f0/a5/d3a92c4904207a2429de4c336b806cb1137ac86d96d053b73a9ed8b3d6ba/datasketches-5.2.0-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:52c53f8c94c48b3f047ab488e9ea41ce8dbf6897c2c1f353cb8a79b22cb4a22b", size = 681874, upload-time = "2025-03-01T07:48:06.356Z" }, + { url = "https://files.pythonhosted.org/packages/c3/cc/b97554d566ca9a3b02645e5e8cb6047e80ee9409bc03f8924ee64e3eba4f/datasketches-5.2.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:2c13fcd9071147a377b587f590887751d85874df677ec5ed72ccdf3f7a19446a", size = 738419, upload-time = "2025-03-01T07:48:08.782Z" }, + { url = "https://files.pythonhosted.org/packages/fe/87/36c48c4af91ab732a09bfe06392df664b30ea780523d719e1be7819ec622/datasketches-5.2.0-cp310-cp310-musllinux_1_2_aarch64.whl", hash = "sha256:62aa8ddfb0e8f4d0e3e1e214e19e0de11b8ea1e34bf7752dfd987e6e9e3b1264", size = 1069353, upload-time = "2025-03-01T07:48:11.088Z" }, + { url = "https://files.pythonhosted.org/packages/4e/f2/abc65ff28286e4997c02d3d83f1236b72595c3e5ba9d3ba39274eab87eed/datasketches-5.2.0-cp310-cp310-musllinux_1_2_x86_64.whl", hash = "sha256:95c2f6b2dea71385bfddf525cc14c9d8bf261f7aa98cbd65155dbaa0764f6cd9", size = 1131149, upload-time = "2025-03-01T07:48:13.893Z" }, + { url = "https://files.pythonhosted.org/packages/fa/4d/be6af6e5cf0bdef5b55d640852a47946a7f044c6fcad989e81619ebb46ab/datasketches-5.2.0-cp310-cp310-win_amd64.whl", hash = "sha256:a8a0500c0de3012fda530e12b6c62d27278f7d6cce21008b9208895dfd337201", size = 506641, upload-time = "2025-03-01T07:48:15.779Z" }, + { url = "https://files.pythonhosted.org/packages/c3/b9/5c69df548b19fc2a20d5990f32023db4483cef43a263d775ba7e08bda00e/datasketches-5.2.0-cp311-cp311-macosx_10_14_x86_64.whl", hash = "sha256:dd8640866011c7dd346d8c9cf9ad0438de16d56788c13b12c96ecb38c5c7df9d", size = 653081, upload-time = "2025-03-01T07:48:17.643Z" }, + { url = "https://files.pythonhosted.org/packages/f5/0e/8baa3ec5ab48408c1ad7d2d4a9ecba07a4a7e16392b8e0d3480fe88b0285/datasketches-5.2.0-cp311-cp311-macosx_11_0_arm64.whl", hash = "sha256:e78a462b397c9a11e876a7fd8198d77d35b8bbe849a8d329c518962bc8463ad1", size = 585248, upload-time = "2025-03-01T07:48:19.778Z" }, + { url = "https://files.pythonhosted.org/packages/fc/99/011c7edd1c7971ca4a5c2e545a99889716d907498e25f96e1055c1a9c49f/datasketches-5.2.0-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:7cde9ce8f42b6a05930dd9b5d0da1b4420d1e8db53faaa41c6732ce7a1670c4a", size = 682144, upload-time = "2025-03-01T07:48:21.663Z" }, + { url = "https://files.pythonhosted.org/packages/ce/27/820dc70c6d4b23fc43187113a9386d4f8702c162f4387f459dfc5063ee63/datasketches-5.2.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:470111ef1b213bc75b3602e14153c3e70a1ebe0fe4b069a7800557685080c477", size = 738715, upload-time = "2025-03-01T07:48:24.085Z" }, + { url = "https://files.pythonhosted.org/packages/94/42/5e8bc5277891797e3207a02ab12db3b2ae7159a45aae917a510c690c5880/datasketches-5.2.0-cp311-cp311-musllinux_1_2_aarch64.whl", hash = "sha256:0208fb7bf40edca3f8cfcd9d1acdb95798abc10b685341b39557f59eb5bd862e", size = 1069861, upload-time = "2025-03-01T07:48:26.419Z" }, + { url = "https://files.pythonhosted.org/packages/62/43/765df9fd6ea2f6fffc1fd757e28e4736568abf9ff1789471d3d8d7dc9ad8/datasketches-5.2.0-cp311-cp311-musllinux_1_2_x86_64.whl", hash = "sha256:bfa6571373244a9e3a4a1f59919b8a41979a997b92be3db0219400d8c6e96d20", size = 1131346, upload-time = "2025-03-01T07:48:28.756Z" }, + { url = "https://files.pythonhosted.org/packages/4a/f4/76a267c596c0a9849241efc91ffb9791fd8b6d818b594d38ce0a322215e7/datasketches-5.2.0-cp311-cp311-win_amd64.whl", hash = "sha256:516e67cd3fd2d14c58c9d2b593f50ac23aff3874406c325ba2589a9c989862ca", size = 507446, upload-time = "2025-03-01T07:48:30.823Z" }, + { url = "https://files.pythonhosted.org/packages/fe/88/ac2dc472e9d054e3edbce9a5d1288bab447fc080d5a0d85984455cf66808/datasketches-5.2.0-cp312-cp312-macosx_10_14_x86_64.whl", hash = "sha256:65b3e2bb6b2083dfdd6a611e4b07d058ca6625f838088b5cb0897335c4580e6d", size = 643447, upload-time = "2025-03-01T07:48:32.844Z" }, + { url = "https://files.pythonhosted.org/packages/00/0c/faca927b0575482d567eda4fa65ffe5ebe1ac04b6c5f0321faeb490d5b8e/datasketches-5.2.0-cp312-cp312-macosx_11_0_arm64.whl", hash = "sha256:bfc80979ccb2e11bf5e93996db7f1f486c9d3ef4a15c18f1f1bc37aaa4b2a038", size = 578736, upload-time = "2025-03-01T07:48:35.427Z" }, + { url = "https://files.pythonhosted.org/packages/ce/6f/9201d36b6775ec8dc3f7d4cc10e32b35b2012683b520d0cf83b0cb674866/datasketches-5.2.0-cp312-cp312-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:a76731c0d876cff25cc2896ef9d80c54167f5aff7170f12b89db6a9b898fc714", size = 675679, upload-time = "2025-03-01T07:48:38.157Z" }, + { url = "https://files.pythonhosted.org/packages/af/87/56bbb0be6d6c49c9b1705f56bad3f4319f8cdbc546f79e811ac88bdfe2cc/datasketches-5.2.0-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:d9aa0c1dad8d00c242718c5dbb9f8cf1876ea099ee55c1c0634a8beecb589164", size = 748356, upload-time = "2025-03-01T07:48:40.344Z" }, + { url = "https://files.pythonhosted.org/packages/c9/37/0c25e113ae8a148201d60050f26efeada54f46e8417bd35082b40355eac3/datasketches-5.2.0-cp312-cp312-musllinux_1_2_aarch64.whl", hash = "sha256:40e6cecb8656b6694e845d9319c5c3aaa083f754e992db6065f18806474685c1", size = 1070320, upload-time = "2025-03-01T07:48:43.218Z" }, + { url = "https://files.pythonhosted.org/packages/0f/43/5ea198ff05ba3c1f904a1118b23f1b36b0c60f16b15af791a3ab4ccee752/datasketches-5.2.0-cp312-cp312-musllinux_1_2_x86_64.whl", hash = "sha256:a673180dc5226bf3c84baeeadf36d9a4f1ee82e679dcd92776651f8ffcc09287", size = 1146824, upload-time = "2025-03-01T07:48:46.841Z" }, + { url = "https://files.pythonhosted.org/packages/8d/fc/1ca195f0fe524e3ffc12ac81c07f3cc3ce144499b4e3ec5e73a52167253f/datasketches-5.2.0-cp312-cp312-win_amd64.whl", hash = "sha256:72cc339cb7775f82516c36fe8e9a70e9f504214a8d0ed06246332f53203b89f3", size = 509067, upload-time = "2025-03-01T07:48:48.784Z" }, + { url = "https://files.pythonhosted.org/packages/62/a7/2b69296c200bd59550cb6ee292d8ba6739ea2d847e10d38452d86120bb45/datasketches-5.2.0-cp313-cp313-macosx_10_14_x86_64.whl", hash = "sha256:a76c998ddf4d39f895b830a3ccc41d1df2d0454c74fc5e53844db119658e71a7", size = 643448, upload-time = "2025-03-01T07:48:50.808Z" }, + { url = "https://files.pythonhosted.org/packages/ed/56/ca425991d21e4b4e4b0a72276a77678201a92d5609acbb27ad7a05ddfce6/datasketches-5.2.0-cp313-cp313-macosx_11_0_arm64.whl", hash = "sha256:6c2be3884b28d24a3be103a5c285902f06eee85c076be57c62c9c5eecbe15d4d", size = 578736, upload-time = "2025-03-01T07:48:53.73Z" }, + { url = "https://files.pythonhosted.org/packages/2b/89/d8ce2f6eab2914a5091360f94fc52cb6f93b1e3852f1aa86dbba9833e20f/datasketches-5.2.0-cp313-cp313-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:9bcbf2687b436593f4e03ba73f57dd72fc6a7414d88db57e91f4c24f6e62ee45", size = 675679, upload-time = "2025-03-01T07:48:55.734Z" }, + { url = "https://files.pythonhosted.org/packages/62/33/351d1f0c700e143217597d29b333c77695db0f0b3757cf3c2b6e8cf58ea7/datasketches-5.2.0-cp313-cp313-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:49e78e52d6a1d63b08a2990873f94bbc9d7427f6907f600af1257f0a9c901b1f", size = 748357, upload-time = "2025-03-01T07:48:57.901Z" }, + { url = "https://files.pythonhosted.org/packages/7d/6c/9ef89caf91c2d2b70fce5038607f2b5d8b09ee14db6bbe89027a8421ed88/datasketches-5.2.0-cp313-cp313-musllinux_1_2_aarch64.whl", hash = "sha256:6be79e382a4b4fe033a7d3de0033fe81b0e01ebc5124ae24f785214517a847a7", size = 1070320, upload-time = "2025-03-01T07:49:01.074Z" }, + { url = "https://files.pythonhosted.org/packages/82/22/a6281d53249af4570b36418de2367114a4ecebd16099e195346641e9d5e8/datasketches-5.2.0-cp313-cp313-musllinux_1_2_x86_64.whl", hash = "sha256:fbd6eab5af078eb65d678bf27fc468097a1c5db07800ac537a75fd983c437e57", size = 1146824, upload-time = "2025-03-01T07:49:04.782Z" }, + { url = "https://files.pythonhosted.org/packages/6f/da/c3feb5eca3d7c43d068069b56f77685c01d1ee67e687490cc6341ec920f1/datasketches-5.2.0-cp313-cp313-win_amd64.whl", hash = "sha256:bf8f1cc1b1c4a35554924e27d6c2872e7b0dc065e2694ec83035ffbc203f17c3", size = 509058, upload-time = "2025-03-01T07:49:06.796Z" }, +] + [[package]] name = "deptry" version = "0.25.1" @@ -5034,6 +5073,7 @@ requires-dist = [ { name = "click", specifier = ">=7.1.1" }, { name = "daft", marker = "extra == 'daft'", specifier = ">=0.7.10" }, { name = "datafusion", marker = "extra == 'datafusion'", specifier = ">=52,<53" }, + { name = "datasketches", marker = "extra == 'datasketches'", specifier = ">=3.4.0" }, { name = "duckdb", marker = "extra == 'duckdb'", specifier = ">=0.5.0" }, { name = "fsspec", specifier = ">=2023.1.0" }, { name = "gcsfs", marker = "extra == 'gcsfs'", specifier = ">=2023.1.0" }, @@ -5069,12 +5109,13 @@ requires-dist = [ { name = "thrift-sasl", marker = "extra == 'hive-kerberos'", specifier = ">=0.4.3" }, { name = "zstandard", specifier = ">=0.13.0" }, ] -provides-extras = ["pyarrow", "pandas", "duckdb", "ray", "bodo", "daft", "polars", "snappy", "hive", "hive-kerberos", "s3fs", "glue", "adlfs", "dynamodb", "bigquery", "sql-postgres", "sql-sqlite", "gcsfs", "rest-sigv4", "hf", "pyiceberg-core", "datafusion", "gcp-auth", "entra-auth", "geoarrow"] +provides-extras = ["pyarrow", "pandas", "duckdb", "ray", "bodo", "daft", "polars", "snappy", "hive", "hive-kerberos", "s3fs", "glue", "adlfs", "dynamodb", "bigquery", "sql-postgres", "sql-sqlite", "gcsfs", "rest-sigv4", "hf", "pyiceberg-core", "datafusion", "gcp-auth", "entra-auth", "geoarrow", "datasketches"] [package.metadata.requires-dev] dev = [ { name = "coverage", extras = ["toml"], specifier = ">=7.4.2" }, { name = "cython", specifier = ">=3.0.0" }, + { name = "datasketches", specifier = ">=3.4.0" }, { name = "deptry", specifier = ">=0.14" }, { name = "docutils", specifier = "!=0.21.post1" }, { name = "fastavro", specifier = "==1.12.2" },