bigframes 2.0.0.dev0__tar.gz → 2.2.0__tar.gz

{bigframes-2.0.0.dev0/third_party/bigframes_vendored/google_cloud_bigquery → bigframes-2.2.0}/LICENSE

@@ -1,3 +1,6 @@
+Files: All files not covered by another license. Notably: the bigframes module,
+tests/*, bigframes_vendored.google_cloud_bigquery module,
+bigframes_vendored.ibis module, and bigframes_vendored.xgboost module.
 
                                  Apache License
                            Version 2.0, January 2004
@@ -200,3 +203,118 @@
    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.
+
+---
+
+Files: For the bigframes_vendored.cpython module.
+
+PYTHON SOFTWARE FOUNDATION LICENSE VERSION 2
+
+1. This LICENSE AGREEMENT is between the Python Software Foundation ("PSF"), and the Individual or Organization ("Licensee") accessing and otherwise using this software ("Python") in source or binary form and its associated documentation.
+2. Subject to the terms and conditions of this License Agreement, PSF hereby grants Licensee a nonexclusive, royalty-free, world-wide license to reproduce, analyze, test, perform and/or display publicly, prepare derivative works, distribute, and otherwise use Python alone or in any derivative version, provided, however, that PSF's License Agreement and PSF's notice of copyright , i.e., "Copyright (c) 2001, 2002, 2003, 2004, 2005, 2006 Python Software Foundation All Rights Reserved" are retained in Python alone or in any derivative version prepared by Licensee.
+3. In the event Licensee prepares a derivative work that is based on or incorporates Python or any part thereof, and wants to make the derivative work available to others as provided herein, then Licensee hereby agrees to include in any such work a brief summary of the changes made to Python.
+4. PSF is making Python available to Licensee on an "AS IS" basis. PSF MAKES NO REPRESENTATIONS OR WARRANTIES, EXPRESS OR IMPLIED. BY WAY OF EXAMPLE, BUT NOT LIMITATION, PSF MAKES NO AND DISCLAIMS ANY REPRESENTATION OR WARRANTY OF MERCHANTABILITY OR FITNESS FOR ANY PARTICULAR PURPOSE OR THAT THE USE OF PYTHON WILL NOT INFRINGE ANY THIRD PARTY RIGHTS.
+5. PSF SHALL NOT BE LIABLE TO LICENSEE OR ANY OTHER USERS OF PYTHON FOR ANY INCIDENTAL, SPECIAL, OR CONSEQUENTIAL DAMAGES OR LOSS AS A RESULT OF MODIFYING, DISTRIBUTING, OR OTHERWISE USING PYTHON, OR ANY DERIVATIVE THEREOF, EVEN IF ADVISED OF THE POSSIBILITY THEREOF.
+6. This License Agreement will automatically terminate upon a material breach of its terms and conditions.
+7. Nothing in this License Agreement shall be deemed to create any relationship of agency, partnership, or joint venture between PSF and Licensee. This License Agreement does not grant permission to use PSF trademarks or trade name in a trademark sense to endorse or promote products or services of Licensee, or any third party.
+8. By copying, installing or otherwise using Python, Licensee agrees to be bound by the terms and conditions of this License Agreement.
+
+---
+
+Files: for the bigframes_vendored.geopandas module.
+
+Copyright (c) 2013-2022, GeoPandas developers.
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+ * Redistributions of source code must retain the above copyright notice, this
+   list of conditions and the following disclaimer.
+ * Redistributions in binary form must reproduce the above copyright notice,
+   this list of conditions and the following disclaimer in the documentation
+   and/or other materials provided with the distribution.
+ * Neither the name of GeoPandas nor the names of its contributors may
+   be used to endorse or promote products derived from this software without
+   specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
+ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
+ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
+ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+---
+
+Files: The bigframes_vendored.pandas module.
+
+BSD 3-Clause License
+
+Copyright (c) 2008-2011, AQR Capital Management, LLC, Lambda Foundry, Inc. and PyData Development Team
+All rights reserved.
+
+Copyright (c) 2011-2023, Open source contributors.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+* Redistributions of source code must retain the above copyright notice, this
+  list of conditions and the following disclaimer.
+
+* Redistributions in binary form must reproduce the above copyright notice,
+  this list of conditions and the following disclaimer in the documentation
+  and/or other materials provided with the distribution.
+
+* Neither the name of the copyright holder nor the names of its
+  contributors may be used to endorse or promote products derived from
+  this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+---
+
+Files: The bigframes_vendored.sklearn module.
+
+BSD 3-Clause License
+
+Copyright (c) 2007-2023 The scikit-learn developers.
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+* Redistributions of source code must retain the above copyright notice, this
+  list of conditions and the following disclaimer.
+
+* Redistributions in binary form must reproduce the above copyright notice,
+  this list of conditions and the following disclaimer in the documentation
+  and/or other materials provided with the distribution.
+
+* Neither the name of the copyright holder nor the names of its
+  contributors may be used to endorse or promote products derived from
+  this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

{bigframes-2.0.0.dev0/bigframes.egg-info → bigframes-2.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: bigframes
-Version: 2.0.0.dev0
+Version: 2.2.0
 Summary: BigQuery DataFrames -- scalable analytics and machine learning with BigQuery
 Home-page: https://github.com/googleapis/python-bigquery-dataframes
 Author: Google LLC
@@ -26,22 +26,23 @@ Requires-Dist: cloudpickle>=2.0.0
 Requires-Dist: fsspec>=2023.3.0
 Requires-Dist: gcsfs>=2023.3.0
 Requires-Dist: geopandas>=0.12.2
-Requires-Dist: google-auth<3.0dev,>=2.15.0
+Requires-Dist: google-auth<3.0,>=2.15.0
 Requires-Dist: google-cloud-bigtable>=2.24.0
 Requires-Dist: google-cloud-pubsub>=2.21.4
 Requires-Dist: google-cloud-bigquery[bqstorage,pandas]>=3.31.0
+Requires-Dist: google-cloud-bigquery-storage<3.0.0,>=2.30.0
 Requires-Dist: google-cloud-functions>=1.12.0
 Requires-Dist: google-cloud-bigquery-connection>=1.12.0
 Requires-Dist: google-cloud-iam>=2.12.1
 Requires-Dist: google-cloud-resource-manager>=1.10.3
 Requires-Dist: google-cloud-storage>=2.0.0
-Requires-Dist: jellyfish<1.1.2,>=0.8.9
 Requires-Dist: numpy>=1.24.0
 Requires-Dist: pandas>=1.5.3
 Requires-Dist: pandas-gbq>=0.26.1
 Requires-Dist: pyarrow>=15.0.2
 Requires-Dist: pydata-google-auth>=1.8.2
 Requires-Dist: requests>=2.27.1
+Requires-Dist: shapely>=1.8.5
 Requires-Dist: sqlglot>=23.6.3
 Requires-Dist: tabulate>=0.9
 Requires-Dist: ipywidgets>=7.7.1
@@ -49,30 +50,31 @@ Requires-Dist: humanize>=4.6.0
 Requires-Dist: matplotlib>=3.7.1
 Requires-Dist: db-dtypes>=1.4.2
 Requires-Dist: atpublic<6,>=2.3
-Requires-Dist: parsy<3,>=2
 Requires-Dist: python-dateutil<3,>=2.8.2
 Requires-Dist: pytz>=2022.7
 Requires-Dist: toolz<2,>=0.11
 Requires-Dist: typing-extensions<5,>=4.5.0
 Requires-Dist: rich<14,>=12.4.4
 Provides-Extra: tests
+Requires-Dist: freezegun; extra == "tests"
+Requires-Dist: pytest-snapshot; extra == "tests"
 Provides-Extra: polars
 Requires-Dist: polars>=1.7.0; extra == "polars"
 Provides-Extra: scikit-learn
 Requires-Dist: scikit-learn>=1.2.2; extra == "scikit-learn"
 Provides-Extra: dev
 Requires-Dist: pytest; extra == "dev"
-Requires-Dist: pytest-mock; extra == "dev"
 Requires-Dist: pre-commit; extra == "dev"
 Requires-Dist: nox; extra == "dev"
 Requires-Dist: google-cloud-testutils; extra == "dev"
 Provides-Extra: all
+Requires-Dist: freezegun; extra == "all"
 Requires-Dist: google-cloud-testutils; extra == "all"
 Requires-Dist: nox; extra == "all"
 Requires-Dist: polars>=1.7.0; extra == "all"
 Requires-Dist: pre-commit; extra == "all"
 Requires-Dist: pytest; extra == "all"
-Requires-Dist: pytest-mock; extra == "all"
+Requires-Dist: pytest-snapshot; extra == "all"
 Requires-Dist: scikit-learn>=1.2.2; extra == "all"
 
 BigQuery DataFrames
@@ -102,6 +104,7 @@ Version 2.0 introduces breaking changes for improved security and performance. K
   ``cloud_function_service_account="default"``. And network ingress now defaults to ``"internal-only"``.
 * **@remote_function Argument Passing:** Arguments other than ``input_types``, ``output_type``, and ``dataset``
   to ``remote_function`` must now be passed using keyword syntax, as positional arguments are no longer supported.
+* **@udf Argument Passing:** Arguments ``dataset`` and ``name`` to ``udf`` are now mandatory.
 * **Endpoint Connections:** Automatic fallback to locational endpoints in certain regions is removed.
 * **LLM Updates (Gemini Integration):** Integrations now default to the ``gemini-2.0-flash-001`` model.
   PaLM2 support has been removed; please migrate any existing PaLM2 usage to Gemini. **Note:** The current default

{bigframes-2.0.0.dev0 → bigframes-2.2.0}/README.rst

@@ -25,6 +25,7 @@ Version 2.0 introduces breaking changes for improved security and performance. K
   ``cloud_function_service_account="default"``. And network ingress now defaults to ``"internal-only"``.
 * **@remote_function Argument Passing:** Arguments other than ``input_types``, ``output_type``, and ``dataset``
   to ``remote_function`` must now be passed using keyword syntax, as positional arguments are no longer supported.
+* **@udf Argument Passing:** Arguments ``dataset`` and ``name`` to ``udf`` are now mandatory.
 * **Endpoint Connections:** Automatic fallback to locational endpoints in certain regions is removed.
 * **LLM Updates (Gemini Integration):** Integrations now default to the ``gemini-2.0-flash-001`` model.
   PaLM2 support has been removed; please migrate any existing PaLM2 usage to Gemini. **Note:** The current default

{bigframes-2.0.0.dev0 → bigframes-2.2.0}/bigframes/_config/bigquery_options.py

@@ -21,7 +21,6 @@ import warnings
 
 import google.api_core.exceptions
 import google.auth.credentials
-import jellyfish
 
 import bigframes.constants
 import bigframes.enums
@@ -37,6 +36,7 @@ UNKNOWN_LOCATION_MESSAGE = "The location '{location}' is set to an unknown value
 
 
 def _get_validated_location(value: Optional[str]) -> Optional[str]:
+    import bigframes._tools.strings
 
     if value is None or value in bigframes.constants.ALL_BIGQUERY_LOCATIONS:
         return value
@@ -53,7 +53,7 @@ def _get_validated_location(value: Optional[str]) -> Optional[str]:
 
     possibility = min(
         bigframes.constants.ALL_BIGQUERY_LOCATIONS,
-        key=lambda item: jellyfish.levenshtein_distance(location, item),
+        key=lambda item: bigframes._tools.strings.levenshtein_distance(location, item),
     )
     # There are many layers before we get to (possibly) the user's code:
     # -> bpd.options.bigquery.location = "us-central-1"

{bigframes-2.0.0.dev0 → bigframes-2.2.0}/bigframes/_config/experiment_options.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from typing import Optional
 import warnings
 
 import bigframes.exceptions as bfe
@@ -26,6 +27,9 @@ class ExperimentOptions:
         self._semantic_operators: bool = False
         self._ai_operators: bool = False
         self._blob: bool = False
+        self._blob_display: bool = True
+        self._blob_display_width: Optional[int] = None
+        self._blob_display_height: Optional[int] = None
 
     @property
     def semantic_operators(self) -> bool:
@@ -67,3 +71,30 @@ class ExperimentOptions:
             )
             warnings.warn(msg, category=bfe.PreviewWarning)
         self._blob = value
+
+    @property
+    def blob_display(self) -> bool:
+        """Whether to display the blob content in notebook DataFrame preview. Default True."""
+        return self._blob_display
+
+    @blob_display.setter
+    def blob_display(self, value: bool):
+        self._blob_display = value
+
+    @property
+    def blob_display_width(self) -> Optional[int]:
+        """Width in pixels that the blob constrained to."""
+        return self._blob_display_width
+
+    @blob_display_width.setter
+    def blob_display_width(self, value: Optional[int]):
+        self._blob_display_width = value
+
+    @property
+    def blob_display_height(self) -> Optional[int]:
+        """Height in pixels that the blob constrained to."""
+        return self._blob_display_height
+
+    @blob_display_height.setter
+    def blob_display_height(self, value: Optional[int]):
+        self._blob_display_height = value

bigframes-2.2.0/bigframes/_tools/__init__.py

@@ -0,0 +1,19 @@
+# Copyright 2025 Google LLC
+#
+# Licensed 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.
+
+"""_tools is a collection of helper functions with minimal dependencies.
+
+Please keep the dependencies used in this subpackage to a minimum to avoid the
+risk of circular dependencies.
+"""

bigframes-2.2.0/bigframes/_tools/strings.py

@@ -0,0 +1,66 @@
+# Copyright 2025 Google LLC
+#
+# Licensed 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.
+
+"""Helper methods for processing strings with minimal dependencies.
+
+Please keep the dependencies used in this subpackage to a minimum to avoid the
+risk of circular dependencies.
+"""
+
+import numpy
+
+
+def levenshtein_distance(left: str, right: str) -> int:
+    """Compute the edit distance between two strings.
+
+    This is the minumum number of substitutions, insertions, deletions
+    to get from left string to right string. See:
+    https://en.wikipedia.org/wiki/Levenshtein_distance
+    """
+    # TODO(tswast): accelerate with numba (if available) if we end up using this
+    # function in contexts other than when raising an exception or there are too
+    # many values to compare even in that context.
+
+    distances0 = numpy.zeros(len(right) + 1)
+    distances1 = numpy.zeros(len(right) + 1)
+
+    # Maximum distance is to drop all characters and then add the other string.
+    distances0[:] = range(len(right) + 1)
+
+    for left_index in range(len(left)):
+        # Calculate distance from distances0 to distances1.
+
+        # Edit distance is to delete (i + 1) chars from left to match empty right
+        distances1[0] = left_index + 1
+        # "ab"
+        for right_index in range(len(right)):
+            left_char = left[left_index]
+            right_char = right[right_index]
+
+            deletion_cost = distances0[right_index + 1] + 1
+            insertion_cost = distances1[right_index] + 1
+            if left_char == right_char:
+                substitution_cost = distances0[right_index]
+            else:
+                substitution_cost = distances0[right_index] + 1
+
+            distances1[right_index + 1] = min(
+                deletion_cost, insertion_cost, substitution_cost
+            )
+
+        temp = distances0
+        distances0 = distances1
+        distances1 = temp
+
+    return distances0[len(right)]

{bigframes-2.0.0.dev0 → bigframes-2.2.0}/bigframes/bigquery/__init__.py

@@ -27,7 +27,12 @@ from bigframes.bigquery._operations.datetime import (
     unix_millis,
     unix_seconds,
 )
-from bigframes.bigquery._operations.geo import st_area, st_difference, st_intersection
+from bigframes.bigquery._operations.geo import (
+    st_area,
+    st_difference,
+    st_distance,
+    st_intersection,
+)
 from bigframes.bigquery._operations.json import (
     json_extract,
     json_extract_array,
@@ -49,6 +54,7 @@ __all__ = [
     # geo ops
     "st_area",
     "st_difference",
+    "st_distance",
     "st_intersection",
     # json ops
     "json_set",

{bigframes-2.0.0.dev0 → bigframes-2.2.0}/bigframes/bigquery/_operations/geo.py

@@ -14,8 +14,11 @@
 
 from __future__ import annotations
 
+from typing import Union
+
+import shapely  # type: ignore
+
 from bigframes import operations as ops
-import bigframes.dtypes
 import bigframes.geopandas
 import bigframes.series
 
@@ -25,7 +28,9 @@ https://cloud.google.com/bigquery/docs/reference/standard-sql/geography_function
 """
 
 
-def st_area(series: bigframes.series.Series) -> bigframes.series.Series:
+def st_area(
+    series: Union[bigframes.series.Series, bigframes.geopandas.GeoSeries],
+) -> bigframes.series.Series:
     """
     Returns the area in square meters covered by the polygons in the input
     `GEOGRAPHY`.
@@ -85,6 +90,10 @@ def st_area(series: bigframes.series.Series) -> bigframes.series.Series:
         4            0.0
         dtype: Float64
 
+    Args:
+        series (bigframes.pandas.Series | bigframes.geopandas.GeoSeries):
+            A series containing geography objects.
+
     Returns:
       bigframes.pandas.Series:
           Series of float representing the areas.
@@ -95,7 +104,12 @@ def st_area(series: bigframes.series.Series) -> bigframes.series.Series:
 
 
 def st_difference(
-    series: bigframes.series.Series, other: bigframes.series.Series
+    series: Union[bigframes.series.Series, bigframes.geopandas.GeoSeries],
+    other: Union[
+        bigframes.series.Series,
+        bigframes.geopandas.GeoSeries,
+        shapely.geometry.base.BaseGeometry,
+    ],
 ) -> bigframes.series.Series:
     """
     Returns a `GEOGRAPHY` that represents the point set difference of
@@ -166,44 +180,23 @@ def st_difference(
         5                                               None
         dtype: geometry
 
-    We can also check difference of single shapely geometries:
-
-        >>> polygon_s1 = bigframes.geopandas.GeoSeries(
-        ...     [
-        ...         Polygon([(0, 0), (10, 0), (10, 10), (0, 0)])
-        ...     ]
-        ... )
-        >>> polygon_s2 = bigframes.geopandas.GeoSeries(
-        ...     [
-        ...         Polygon([(4, 2), (6, 2), (8, 6), (4, 2)])
-        ...     ]
-        ... )
-
-        >>> polygon_s1
-        0    POLYGON ((0 0, 10 0, 10 10, 0 0))
-        dtype: geometry
-
-        >>> polygon_s2
-        0    POLYGON ((4 2, 6 2, 8 6, 4 2))
-        dtype: geometry
-
-        >>> bbq.st_difference(polygon_s1, polygon_s2)
-        0    POLYGON ((0 0, 10 0, 10 10, 0 0), (8 6, 6 2, 4...
-        dtype: geometry
-
     Additionally, we can check difference of a GeoSeries against a single shapely geometry:
 
-        >>> bbq.st_difference(s1, polygon_s2)
-        0    POLYGON ((0 0, 2 2, 0 2, 0 0))
-        1                              None
-        2                              None
-        3                              None
-        4                              None
+        >>> polygon = Polygon([(0, 0), (10, 0), (10, 10), (0, 0)])
+        >>> bbq.st_difference(s1, polygon)
+        0    POLYGON ((1.97082 2.00002, 0 2, 0 0, 1.97082 2...
+        1    POLYGON ((1.97082 2.00002, 0 2, 0 0, 1.97082 2...
+        2                             GEOMETRYCOLLECTION EMPTY
+        3                    LINESTRING (0.99265 1.00781, 0 2)
+        4                                          POINT (0 1)
         dtype: geometry
 
     Args:
-        other (bigframes.series.Series or geometric object):
-            The GeoSeries (elementwise) or geometric object to find the difference to.
+        series (bigframes.pandas.Series | bigframes.geopandas.GeoSeries):
+            A series containing geography objects.
+        other (bigframes.pandas.Series | bigframes.geopandas.GeoSeries | shapely.Geometry):
+            The series or geometric object to subtract from the geography
+            objects in ``series``.
 
     Returns:
         bigframes.series.Series:
@@ -213,8 +206,90 @@ def st_difference(
     return series._apply_binary_op(other, ops.geo_st_difference_op)
 
 
+def st_distance(
+    series: Union[bigframes.series.Series, bigframes.geopandas.GeoSeries],
+    other: Union[
+        bigframes.series.Series,
+        bigframes.geopandas.GeoSeries,
+        shapely.geometry.base.BaseGeometry,
+    ],
+    *,
+    use_spheroid: bool = False,
+) -> bigframes.series.Series:
+    """
+    Returns the shortest distance in meters between two non-empty
+    ``GEOGRAPHY`` objects.
+
+    **Examples:**
+
+        >>> import bigframes as bpd
+        >>> import bigframes.bigquery as bbq
+        >>> import bigframes.geopandas
+        >>> from shapely.geometry import Polygon, LineString, Point
+        >>> bpd.options.display.progress_bar = None
+
+    We can check two GeoSeries against each other, row by row.
+
+        >>> s1 = bigframes.geopandas.GeoSeries(
+        ...    [
+        ...        Point(0, 0),
+        ...        Point(0.00001, 0),
+        ...        Point(0.00002, 0),
+        ...    ],
+        ... )
+        >>> s2 = bigframes.geopandas.GeoSeries(
+        ...    [
+        ...        Point(0.00001, 0),
+        ...        Point(0.00003, 0),
+        ...        Point(0.00005, 0),
+        ...    ],
+        ... )
+
+        >>> bbq.st_distance(s1, s2, use_spheroid=True)
+        0    1.113195
+        1     2.22639
+        2    3.339585
+        dtype: Float64
+
+    We can also calculate the distance of each geometry and a single shapely geometry:
+
+        >>> bbq.st_distance(s2, Point(0.00001, 0))
+        0         0.0
+        1    2.223902
+        2    4.447804
+        dtype: Float64
+
+    Args:
+        series (bigframes.pandas.Series | bigframes.geopandas.GeoSeries):
+            A series containing geography objects.
+        other (bigframes.pandas.Series | bigframes.geopandas.GeoSeries | shapely.Geometry):
+            The series or geometric object to calculate the distance in meters
+            to form the geography objects in ``series``.
+        use_spheroid (optional, default ``False``):
+            Determines how this function measures distance. If ``use_spheroid``
+            is False, the function measures distance on the surface of a perfect
+            sphere. If ``use_spheroid`` is True, the function measures distance
+            on the surface of the `WGS84 spheroid
+            <https://cloud.google.com/bigquery/docs/geospatial-data>`_. The
+            default value of ``use_spheroid`` is False.
+
+    Returns:
+        bigframes.pandas.Series:
+            The Series (elementwise) of the smallest distance between
+            each aligned geometry with other.
+    """
+    return series._apply_binary_op(
+        other, ops.GeoStDistanceOp(use_spheroid=use_spheroid)
+    )
+
+
 def st_intersection(
-    series: bigframes.series.Series, other: bigframes.series.Series
+    series: Union[bigframes.series.Series, bigframes.geopandas.GeoSeries],
+    other: Union[
+        bigframes.series.Series,
+        bigframes.geopandas.GeoSeries,
+        shapely.geometry.base.BaseGeometry,
+    ],
 ) -> bigframes.series.Series:
     """
     Returns a `GEOGRAPHY` that represents the point set intersection of the two
@@ -284,18 +359,20 @@ def st_intersection(
 
     We can also do intersection of each geometry and a single shapely geometry:
 
-        >>> bbq.st_intersection(s1, bigframes.geopandas.GeoSeries([Polygon([(0, 0), (1, 1), (0, 1)])]))
+        >>> bbq.st_intersection(s1, Polygon([(0, 0), (1, 1), (0, 1)]))
         0    POLYGON ((0 0, 0.99954 1, 0 1, 0 0))
-        1                                    None
-        2                                    None
-        3                                    None
-        4                                    None
+        1    POLYGON ((0 0, 0.99954 1, 0 1, 0 0))
+        2             LINESTRING (0 0, 0.99954 1)
+        3                GEOMETRYCOLLECTION EMPTY
+        4                             POINT (0 1)
         dtype: geometry
 
     Args:
-        other (GeoSeries or geometric object):
-            The Geoseries (elementwise) or geometric object to find the
-            intersection with.
+        series (bigframes.pandas.Series | bigframes.geopandas.GeoSeries):
+            A series containing geography objects.
+        other (bigframes.pandas.Series | bigframes.geopandas.GeoSeries | shapely.Geometry):
+            The series or geometric object to intersect with the geography
+            objects in ``series``.
 
     Returns:
         bigframes.geopandas.GeoSeries:

{bigframes-2.0.0.dev0 → bigframes-2.2.0}/bigframes/blob/_functions.py

@@ -69,7 +69,7 @@ class TransformFunction:
     def _create_udf(self):
         """Create Python UDF in BQ. Return name of the UDF."""
         udf_name = str(
-            self._session._loader._storage_manager.generate_unique_resource_id()
+            self._session._anon_dataset_manager.generate_unique_resource_id()
         )
 
         func_body = inspect.getsource(self._func)
@@ -102,6 +102,9 @@ AS r\"\"\"
     def udf(self):
         """Create and return the UDF object."""
         udf_name = self._create_udf()
+
+        # TODO(b/404605969): remove cleanups when UDF fixes dataset deletion.
+        self._session._function_session._update_temp_artifacts(udf_name, "")
         return self._session.read_gbq_function(udf_name)